Update model card.

README.md

@@ -20,12 +20,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the tweet_eval dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2519
-- Accuracy: 0.9212
-- Precision: 0.9333
-- Recall: 0.7424
-- F1: 0.8138
-- Auroc: 0.9510
+- Loss: 0.6404
+- Accuracy: 0.6529
+- Precision: 0.8110
+- Recall: 0.6529
+- F1: 0.6507
+- Auroc: 0.9184
 
 ## Model description
 
@@ -56,10 +56,8 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1     | Auroc  |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|:------:|
-| 0.4845        | 2.5   | 250  | 0.4243          | 0.8061   | 0.2000    | 0.0500 | 0.0800 | 0.9209 |
-| 0.2692        | 5.0   | 500  | 0.2946          | 0.9151   | 0.9333    | 0.7091 | 0.7871 | 0.9469 |
-| 0.192         | 7.5   | 750  | 0.2599          | 0.9151   | 0.9333    | 0.7091 | 0.7871 | 0.9500 |
-| 0.1502        | 10.0  | 1000 | 0.2519          | 0.9212   | 0.9333    | 0.7424 | 0.8138 | 0.9510 |
+| 0.7228        | 0.55  | 500  | 0.7232          | 0.6030   | 0.5625    | 0.6030 | 0.5760 | 0.8937 |
+| 0.64          | 1.1   | 1000 | 0.6404          | 0.6529   | 0.8110    | 0.6529 | 0.6507 | 0.9184 |
 
 
 ### Framework versions

config.json

@@ -9,26 +9,24 @@
   "dropout": 0.1,
   "hidden_dim": 3072,
   "id2label": {
-    "0": "violence",
-    "1": "gender",
-    "2": "race",
-    "3": "sexual_orientation",
-    "4": "religion"
+    "0": "anger",
+    "1": "joy",
+    "2": "optimism",
+    "3": "sadness"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "gender": 1,
-    "race": 2,
-    "religion": 4,
-    "sexual_orientation": 3,
-    "violence": 0
+    "anger": 0,
+    "joy": 1,
+    "optimism": 2,
+    "sadness": 3
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
   "n_heads": 12,
   "n_layers": 6,
   "pad_token_id": 0,
-  "problem_type": "multi_label_classification",
+  "problem_type": "single_label_classification",
   "qa_dropout": 0.1,
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,

export/onnx/config.json

@@ -9,26 +9,24 @@
   "dropout": 0.1,
   "hidden_dim": 3072,
   "id2label": {
-    "0": "violence",
-    "1": "gender",
-    "2": "race",
-    "3": "sexual_orientation",
-    "4": "religion"
+    "0": "anger",
+    "1": "joy",
+    "2": "optimism",
+    "3": "sadness"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "gender": 1,
-    "race": 2,
-    "religion": 4,
-    "sexual_orientation": 3,
-    "violence": 0
+    "anger": 0,
+    "joy": 1,
+    "optimism": 2,
+    "sadness": 3
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
   "n_heads": 12,
   "n_layers": 6,
   "pad_token_id": 0,
-  "problem_type": "multi_label_classification",
+  "problem_type": "single_label_classification",
   "qa_dropout": 0.1,
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,

export/onnx/model.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1223642bb3128a04fab445eb2b87cd4c571ec315637acbb93b660cd942656ff0
-size 267964940
+oid sha256:8e7e47d4b2db0a1b4604cfbfd3f4505bf98c478556bed71735c89654a600275b
+size 267961864

export/torchscript/model.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a1c0c8ebcb7d7b113e56f33f84f64e283303a2d939bcf1a134c66802cd75fe06
-size 268006333
+oid sha256:82fde495d92087ebe2f13180eff62dbaf912edcadefaff21c28798dfd684b381
+size 268003261

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1cc767fc7f1f76694a71152a0eed8643bbe52e5f642616ab200de87988f3b3e9
-size 267841796
+oid sha256:6d754ee892cceb8b16d6f2ff45bd15a754375f5a741fc97d5ff7ed657d2aecd8
+size 267838720

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ebfaeb725b056cf6d3a5a10ab103fe1149f27e538107b04a2862d8e0fd1441f6
+oid sha256:a9f3d8455d06dfa06cbae98a26fceafd4199b8e74d7fb2af3666be8918c88e10
 size 4728