tasinhoque commited on
Commit
128b343
1 Parent(s): 646ba82

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -0
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