Pytorch implementation of RNN, CNN, BiGRU and LSTM for text classifcation
Model | Train acc | Val acc | Test acc |
---|---|---|---|
RNN | 0.79435 | 0.39363 | 0.41181 |
CNN | 0.87247 | 0.45000 | 0.43954 |
Bidirectional GRU | 0.62176 | 0.47636 | 0.47591 |
LSTM with attention | 0.54752 | 0.43363 | 0.42594 |