eli
eli

Reputation: 35

training and validation accuracy and loss for text classification

I am doing text classification with 3 classes, after training my model, the image below is my model accuracy and loss results, it means my models doesn't learn? enter image description here

Upvotes: 0

Views: 323

Answers (1)

Dan Ganea
Dan Ganea

Reputation: 550

It seems like your model is overfitting on the training set. This means it learns specific features of the training set that cannot be generalized to your validation set and consequently data it has not been trained on.

If you are using a neural network, investigate in using techniques that alleviate this like dropout and regularization. Some ideas around this can be seen here.

Upvotes: 1

Related Questions