How do I know and tune the batch size while using patternnet for training a neural network classification model
16 views (last 30 days)
Show older comments
So I tried using the patternet to classify the type of wines using the inbult wines dataset. I was wondering, if there was an option to tune the parameter such as batch size and etc? I know how to change the training fucntion and cost function but I'm unable to figure out how to tweak or tune the batch size. Any help would be appreciated.
0 Comments
Answers (1)
Sahil Jain
on 19 Oct 2021
Hi. As of 2021b, shallow networks such as "patternnet" do not support minibatch training. The training function "traingd" uses the batch gradient descent algorithm and uses the entire dataset for each epoch. For minibatch training, consider using a deep neural network with the "trainNetwork" function. The Create Simple Image Classification Network documentation page will help you get started.
0 Comments
See Also
Categories
Find more on Image Data Workflows in Help Center and File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!