Commit
·
cc4ae90
1
Parent(s):
690fcfe
Update README.md
Browse files
README.md
CHANGED
@@ -1,5 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
1 |
Overview
|
2 |
-
|
3 |
DataSet: dair-ai/emotion
|
4 |
|
5 |
Training:
|
@@ -8,10 +13,3 @@ After third epoch, accuracy: 0.9360
|
|
8 |
|
9 |
Test:
|
10 |
accuracy: 0.9265
|
11 |
-
|
12 |
-
Sentiments:
|
13 |
-
sadness (0), joy (1), love (2), anger (3), fear (4), surprise (5)
|
14 |
-
|
15 |
-
Use Colab Notebook for step by step guidance, which is in "Files and Versions" section.
|
16 |
-
|
17 |
-
|
|
|
1 |
+
This fine tunned model will do sentiment analysis,based on 6 sentiments -
|
2 |
+
sadness (0), joy (1), love (2), anger (3), fear (4), surprise (5)
|
3 |
+
|
4 |
+
Download Colab Notebook "FineTunnedBertBaseModel_Use.ipynb" for step by step guidance, which is in "Files and Versions" section.
|
5 |
+
|
6 |
Overview
|
7 |
+
BaseModel : "bert-base-uncased"
|
8 |
DataSet: dair-ai/emotion
|
9 |
|
10 |
Training:
|
|
|
13 |
|
14 |
Test:
|
15 |
accuracy: 0.9265
|
|
|
|
|
|
|
|
|
|
|
|
|
|