I'm a beginner trying to put together my first project. I had a song classification project in mind, but since I would be manually labeling, I could only reasonably put together about 1000 songs, or 60 hours of music.

I would be classifying with several classes, so it's possible that one class would have as few as 50-100 songs in the training set- this seems like too few! Is there a general rule of thumb for how much data is needed to train a neural network to give it a shot at working?

Edit: I was thinking of using a vanilla LSTM. The input features will have dimension 39, output dimension 6, my first attempt for hidden layer dimension would be 100.

  • 2
    $\begingroup$ This isn't really answerable because not all tasks are easy, and different network architectures and hyperparameter selections will improve/hurt different models in different ways. $\endgroup$
    – Sycorax
    Aug 1 '16 at 13:27
  • $\begingroup$ At a minimum, you need to specify your network structure & how many links there will be to train. $\endgroup$ Aug 1 '16 at 13:29
  • $\begingroup$ Minimum viable dataset might be what you wanted. $\endgroup$ Jan 16 at 12:08

It really depends on your dataset, and network architecture. One rule of thumb I have read (2) was a few thousand samples per class for the neural network to start to perform very well.

In practice, people try and see. It's not rare to find studies showing decent results with a training set smaller than 1000 samples.

A good way to roughly assess to what extent it could be beneficial to have more training samples is to plot the performance of the neural network based against the size of the training set, e.g. from (1):

enter image description here

  • $\begingroup$ Let me make a simple strongly related question. I have a (full) sample of 68 observations only, one target variable and 33 eligible predictors (trendless economic time series). Try to predict the target with neural networks, even in simple specifications, is a waste of time? Or can make sense? More in general maybe you can give me an opinion there stats.stackexchange.com/questions/491065/… it would be appreciated. $\endgroup$
    – markowitz
    Oct 9 '20 at 8:35

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.