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.