This fine tunned model will do sentiment analysis,based on 6 sentiments - | |
sadness (0), joy (1), love (2), anger (3), fear (4), surprise (5) | |
Download Colab Notebook "FineTunnedBertBaseModel_Use.ipynb" for step by step guidance, which is in "Files and Versions" section. | |
Overview | |
BaseModel : "bert-base-uncased" | |
DataSet: dair-ai/emotion | |
Training: | |
After first epoch, accuracy: 0.6497 | |
After third epoch, accuracy: 0.9360 | |
Test: | |
accuracy: 0.9265 | |
hello | |