Overview | |
Model : "bert-base-uncased" | |
DataSet: dair-ai/emotion | |
Training: | |
After first epoch, accuracy: 0.6497 | |
After third epoch, accuracy: 0.9360 | |
Test: | |
accuracy: 0.9265 | |
Sentiments: | |
sadness (0), joy (1), love (2), anger (3), fear (4), surprise (5) | |
Use Colab Notebook for step by step guidance, which is in "Files and Versions" section. | |