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

22 次查看(过去 30 天)
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.

回答(1 个)

Sahil Jain
Sahil Jain 2021-10-19
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.

类别

Help CenterFile Exchange 中查找有关 Deep Learning Toolbox 的更多信息

产品


版本

R2021a

Community Treasure Hunt

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

Start Hunting!

Translated by