NeuroNER icon indicating copy to clipboard operation
NeuroNER copied to clipboard

Generalization With Neuro-NER and overfiting disease

Open Parvez-Khan-1 opened this issue 7 years ago • 4 comments

Hello Frank, I used Neuro-ner to solve one of my problem , it working fine but looks like it suffered with over fitting disease of ML.It showing 99.85% on train_set , 100.00% on validation_set and 82.73% on test_set. I need your suggestion to get rid of this problem so that my model is a generalized model which will perform good on unseen data also, looking forward to hear from you.

Thanks ParvezKhan

Parvez-Khan-1 avatar Dec 18 '17 14:12 Parvez-Khan-1

Increasing dropout values may help you.

grafael avatar Jan 03 '18 01:01 grafael

If i increase dropout rate so, should i also need to increase learning rate?. Currently i have 0.5 drop out rate and 0.005 learning rate.

Parvez-Khan-1 avatar Jan 03 '18 10:01 Parvez-Khan-1

No, as far as I know, changing the learning rate will not prevent the overfitting.

grafael avatar Jan 03 '18 12:01 grafael

Hi ParvezKhan,

In order to address over-fitting, there is really two major things you can do: regularization (increase dropout) and get more data.

Is there anyway you can gather more data for your task? Or perhaps, use your trained model to tag some unlabeled text and bootstrap the process like that.

JohnGiorgi avatar Jan 13 '18 14:01 JohnGiorgi