Downloads · 30 days
0
TarunSinghal/bert-base-uncased-SetimentAnalysis
bert-base-uncased-SetimentAnalysis is a text classification model from TarunSinghal. Use it when you need a label for a piece of text.
This fine tunned model will do sentiment analysis,based on 6 sentiments - sadness (0), joy (1), love (2), anger (3), fear (4), surprise (5)
Downloads · 30 days
0
Access
Public
Updated Jun 14, 2023
Repo size
1.3 GB
Likes
2
Public
Click a slice to open those files.
.data-00000-of-000011.3 GB · 100%
From the Hugging Face model README
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 and run Colab Notebook "FineTunnedBertBaseModel_Use.ipynb" for step by step guidance, which is in "Files and Versions" section. When you run above google colab file you will require following 3 files 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
For any contribution or discussion please let me know in Discussion section.