search

rasbt / musicmood

A machine learning approach to classify songs by mood.
GNU General Public License v3.0
410 stars 107 forks source link

overfitting #2

Open Motorrat opened 8 years ago

Motorrat commented 8 years ago

Sebastian, isn't your model overfitting as the recall falls sharply when you going from train to test data.

rasbt commented 8 years ago

Yup, I agree! I got more stable results with regularized logistic regression but couldn't find a way to get a lower train/test discrepancy with any of the naive Bayes models. I started to collect more data but never got around re-training the models, something that's still on my to-do list for some point in future!