minibatchsize on lstm network

7 次查看(过去 30 天)
NGR MNFD
NGR MNFD 2021-8-1
回答: Meet 2024-8-21
HI all...
I have a training set (51*1cell) and test set (13*1cell).and i want to rain with lstm network so i don not know What a suitable number for minibatchsize?

回答(1 个)

Meet
Meet 2024-8-21
Hi NGR,
Based on my understanding of your question, you have a small dataset of sequences and are seeking to determine the optimal mini-batch size for training an LSTM network to ensure efficient learning and strong model performance. Given the relatively small size of your dataset (51 training samples), opting for a smaller mini-batch size might be more suitable. This approach allows the model to update more frequently and can enhance learning from limited data.
For your dataset, starting with a mini-batch size between 1 and 8 could be beneficial, as it permits more frequent updates. You can experiment with these smaller sizes and gradually increase them to observe how they impact your model's performance and training duration.
You can refer to this article for more detailed information:
I hope this information is helpful.
Thank you.

类别

Help CenterFile Exchange 中查找有关 Statistics and Machine Learning Toolbox 的更多信息

产品


版本

R2019a

Community Treasure Hunt

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

Start Hunting!

Translated by