Avnish Kumar
Update README.md
0b1236c
|
raw
history blame
453 Bytes
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