11 August 2021 4 339 Report

Considering that LSTM are good for time series sequences, I have trained LSTM for multiclass classification task in Pytorch environment, several parameters of LSTM model were tuned including number of layers, no of cells, adding dropout, regularization, training at different batch sizes and different number of epochs. But still I am not getting the best performance, the maximum accuracy I could get on my validation dataset is 80 percent while the training accuracy of 92% at last epoch. Training accuracy also saturates after certain no of epochs and does not improve. (Loss curve attached)

Not sure what's wrong either underfitting, overfitting or my dataset have too much variance.

I will be very thankful if someone can give some guidance.

More Hamza Azam's questions See All
Similar questions and discussions