How do I know and tune the batch size while using patternnet for training a neural network classification model

25 views (last 30 days)
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.

Answers (1)

Sahil Jain
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.

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!