tasinhoque
commited on
Commit
•
128b343
1
Parent(s):
646ba82
Update README.md
Browse files
README.md
CHANGED
@@ -38,6 +38,30 @@ After that, the weights were loaded in a new environment and another epoch of tr
|
|
38 |
As the performance decreased in the fifth epoch, further training was discontinued.
|
39 |
|
40 |
After the 4th epoch, the model achieved a Macro F1 score of 53% on the test set, but the fifth epoch reduced the performance.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
41 |
|
42 |
## Intended uses & limitations
|
43 |
|
|
|
38 |
As the performance decreased in the fifth epoch, further training was discontinued.
|
39 |
|
40 |
After the 4th epoch, the model achieved a Macro F1 score of 53% on the test set, but the fifth epoch reduced the performance.
|
41 |
+
The model on commit "5b532728cef22ca9e9bacc8ff9f5687654d36bf3" attains the following scores on the test set:
|
42 |
+
- Accuracy: 0.4271236410539893
|
43 |
+
- Precision: 0.5101494353184485
|
44 |
+
- Recall: 0.5763722014150806
|
45 |
+
- macro-F1: 0.5297380709491947
|
46 |
+
|
47 |
+
Load the model using this syntax:
|
48 |
+
```py
|
49 |
+
import os
|
50 |
+
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
51 |
+
|
52 |
+
os.environ["TOKENIZERS_PARALLELISM"] = "FALSE"
|
53 |
+
|
54 |
+
model_name = "tasinhoque/text-classification-goemotions"
|
55 |
+
commit = "5b532728cef22ca9e9bacc8ff9f5687654d36bf3"
|
56 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name, revision=commit)
|
57 |
+
|
58 |
+
model = AutoModelForSequenceClassification.from_pretrained(
|
59 |
+
model_name,
|
60 |
+
num_labels=n_emotion,
|
61 |
+
problem_type="multi_label_classification",
|
62 |
+
revision=commit
|
63 |
+
).to(device)
|
64 |
+
```
|
65 |
|
66 |
## Intended uses & limitations
|
67 |
|