File size: 1,457 Bytes
63644eb cc4ae90 f8dc05f 7b18ee5 cc4ae90 87d5ccb 26d7d1f 87d5ccb 26d7d1f 87d5ccb ed5d4e8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
---
datasets:
- dair-ai/emotion
- SetFit/emotion
language:
- en
pipeline_tag: text-classification
tags:
- Transformers
- Text Classification
- bert-base-uncased
- emotion-classification
---
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
1. model.index
2. checkpoint
3. model.data-00000-of-00001
Create a folder on you google drive name, folder name should be "model2.1" and save all these 3 files in it.
If you want to change folder name instead of model2.1, you want folder name should be "sentimentXYZ" then you have to change
line "classifier_2.load_weights('/content/drive/MyDrive/FineTunning2/model2.1/model')" in "FineTunnedBertBaseModel_Use.ipynb" to
"classifier_2.load_weights('/content/drive/MyDrive/FineTunning2/setimentXYZ/model')" this is the path of your save weights on
In the following path '/content/drive/MyDrive/FineTunning2/model2.1/model'
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. |