How do I know and tune the batch size while using patternnet for training a neural network classification model
20 次查看(过去 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.
0 个评论
回答(1 个)
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.
0 个评论
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!