File size: 453 Bytes
cc4ae90 7b18ee5 cc4ae90 87d5ccb 26d7d1f 87d5ccb 26d7d1f 87d5ccb 0b1236c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | 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
|