TarunSinghal's picture
Update README.md
698a1d8
|
raw
history blame
340 Bytes

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.