Update README.md
Browse files
README.md
CHANGED
|
@@ -2,9 +2,6 @@
|
|
| 2 |
license: apache-2.0
|
| 3 |
datasets:
|
| 4 |
- amazon_polarity
|
| 5 |
-
metrics:
|
| 6 |
-
- accuracy
|
| 7 |
-
- f1
|
| 8 |
base_model: distilbert-base-uncased
|
| 9 |
model-index:
|
| 10 |
- name: distilbert-base-uncased-finetuned-emotion-balanced
|
|
@@ -18,7 +15,7 @@ model-index:
|
|
| 18 |
args: default
|
| 19 |
metrics:
|
| 20 |
- type: accuracy
|
| 21 |
-
value: 0.
|
| 22 |
name: Accuracy
|
| 23 |
- type: loss
|
| 24 |
value: 0.119
|
|
@@ -39,41 +36,47 @@ model-index:
|
|
| 39 |
value: 0.94112
|
| 40 |
name: Accuracy
|
| 41 |
verified: true
|
| 42 |
-
verifyToken:
|
|
|
|
| 43 |
- type: precision
|
| 44 |
value: 0.9321570625232675
|
| 45 |
name: Precision
|
| 46 |
verified: true
|
| 47 |
-
verifyToken:
|
|
|
|
| 48 |
- type: recall
|
| 49 |
value: 0.95149
|
| 50 |
name: Recall
|
| 51 |
verified: true
|
| 52 |
-
verifyToken:
|
|
|
|
| 53 |
- type: auc
|
| 54 |
value: 0.9849019044624999
|
| 55 |
name: AUC
|
| 56 |
verified: true
|
| 57 |
-
verifyToken:
|
|
|
|
| 58 |
- type: f1
|
| 59 |
value: 0.9417243188138998
|
| 60 |
name: F1
|
| 61 |
verified: true
|
| 62 |
-
verifyToken:
|
|
|
|
| 63 |
- type: loss
|
| 64 |
value: 0.16342754662036896
|
| 65 |
name: loss
|
| 66 |
verified: true
|
| 67 |
-
verifyToken:
|
|
|
|
| 68 |
---
|
| 69 |
|
| 70 |
# distilbert-sentiment
|
| 71 |
|
| 72 |
This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on a subset of the [amazon-polarity dataset](https://huggingface.co/datasets/amazon_polarity).
|
| 73 |
It achieves the following results on the evaluation set:
|
| 74 |
-
- Loss: 0.
|
| 75 |
-
- Accuracy: 0.
|
| 76 |
-
- F1_score: 0.
|
| 77 |
|
| 78 |
## Model description
|
| 79 |
|
|
@@ -115,10 +118,12 @@ The following hyperparameters were used during training:
|
|
| 115 |
|
| 116 |
| key | value |
|
| 117 |
| --- | ----- |
|
| 118 |
-
| eval_loss | 0.
|
| 119 |
-
| eval_accuracy | 0.
|
| 120 |
-
|
|
| 121 |
-
|
|
|
|
|
|
|
| 122 |
### Framework versions
|
| 123 |
|
| 124 |
- Transformers 4.34.0
|
|
|
|
| 2 |
license: apache-2.0
|
| 3 |
datasets:
|
| 4 |
- amazon_polarity
|
|
|
|
|
|
|
|
|
|
| 5 |
base_model: distilbert-base-uncased
|
| 6 |
model-index:
|
| 7 |
- name: distilbert-base-uncased-finetuned-emotion-balanced
|
|
|
|
| 15 |
args: default
|
| 16 |
metrics:
|
| 17 |
- type: accuracy
|
| 18 |
+
value: 0.94112
|
| 19 |
name: Accuracy
|
| 20 |
- type: loss
|
| 21 |
value: 0.119
|
|
|
|
| 36 |
value: 0.94112
|
| 37 |
name: Accuracy
|
| 38 |
verified: true
|
| 39 |
+
verifyToken: >-
|
| 40 |
+
eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiMzlmMzdhYjNmN2U0NDBkM2U5ZDgwNzc3YjE1OGE4MWUxMDY1N2U0ODc0YzllODE5ODIyMzdkOWFhNzVjYmI5MyIsInZlcnNpb24iOjF9.3nlcLa4IpPQtklp7_U9XzC__Q_JVf_cWs6JVVII8trhX5zg_q9HEyQOQs4sRf6O-lIJg8zb3mgobZDJShuSJAQ
|
| 41 |
- type: precision
|
| 42 |
value: 0.9321570625232675
|
| 43 |
name: Precision
|
| 44 |
verified: true
|
| 45 |
+
verifyToken: >-
|
| 46 |
+
eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiZjI2MDY4NGNlYjhjMGMxODBiNTc2ZjM5YzY1NjkxNTU4MDA2ZDIyY2QyZjUyZmE4YWY0N2Y1ODU5YTc2ZDM0NiIsInZlcnNpb24iOjF9.egEikTa2UyHV6SAGkHJKaa8FRwGHoZmJRCmqUQaJqeF5yxkz2V-WeCHoWDrCXsHCbXEs8UhLlyo7Lr83BPfkBg
|
| 47 |
- type: recall
|
| 48 |
value: 0.95149
|
| 49 |
name: Recall
|
| 50 |
verified: true
|
| 51 |
+
verifyToken: >-
|
| 52 |
+
eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiM2E3M2Y3MDU4ZTM2YjdlZjQ0NTY3NGYwMmQ3NTk5ZmZkZWUwZWZiZDZjNjk2ZWE5MmY4MmZiM2FmN2U2M2QyNCIsInZlcnNpb24iOjF9.4VNbiWRmSee4cxuIZ5m7bN30i4BpK7xtHQ1BF8AuFIXkWQgzOmGdX35bLhLGWW8KL3ClA4RDPVBKYCIrw0YUBw
|
| 53 |
- type: auc
|
| 54 |
value: 0.9849019044624999
|
| 55 |
name: AUC
|
| 56 |
verified: true
|
| 57 |
+
verifyToken: >-
|
| 58 |
+
eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiYTkwODk2ZTUwOTViNjBhYTU0ODk1MDA3MDY1NDkyZDc2YmRlNTQzNDE3YmE3YTVkYjNhN2JmMDAxZWQ0NjUxZSIsInZlcnNpb24iOjF9.YEr6OhqOL7QnqYqjUTQFMdkgU_uS1-vVnkJtn_-1UwSoX754UV_bL9S9KSH3DX4m5QFoRXdZxfeOocm1JbzaCA
|
| 59 |
- type: f1
|
| 60 |
value: 0.9417243188138998
|
| 61 |
name: F1
|
| 62 |
verified: true
|
| 63 |
+
verifyToken: >-
|
| 64 |
+
eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiMzIyMmViNTQ3ZGU0M2I5ZmRjOGI1OWMwZGEwYmE5OGU5YTZlZTkzZjdkOTQ4YzJmOTc2MDliMDY4NDQ1NGRlNyIsInZlcnNpb24iOjF9.p05MGHTfHTAzp4u-qfiIn6Zmh5c3TW_uwjXWgbb982pL_oCILQb6jFXqhPpWXL321fPye7qaUVbGhcTJd8sdCA
|
| 65 |
- type: loss
|
| 66 |
value: 0.16342754662036896
|
| 67 |
name: loss
|
| 68 |
verified: true
|
| 69 |
+
verifyToken: >-
|
| 70 |
+
eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiNzgxMDc4M2IxYjhkNjRhZmYyNzY1MTNkNzhmYjk2NmU1NjFiOTk1NDIzNzI1ZGU3MDYyYjQ2YmQ1NTI2N2NhMyIsInZlcnNpb24iOjF9.Zuf0nzn8XdvwRChKtE9CwJ0pgpc6Zey6oTR3jRiSkvNY2sNbo2bvAgFimGzgGYkDvRvYkTCXzCyxdb27l3QnAg
|
| 71 |
---
|
| 72 |
|
| 73 |
# distilbert-sentiment
|
| 74 |
|
| 75 |
This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on a subset of the [amazon-polarity dataset](https://huggingface.co/datasets/amazon_polarity).
|
| 76 |
It achieves the following results on the evaluation set:
|
| 77 |
+
- Loss: 0.163
|
| 78 |
+
- Accuracy: 0.941
|
| 79 |
+
- F1_score: 0.942
|
| 80 |
|
| 81 |
## Model description
|
| 82 |
|
|
|
|
| 118 |
|
| 119 |
| key | value |
|
| 120 |
| --- | ----- |
|
| 121 |
+
| eval_loss | 0.1634 |
|
| 122 |
+
| eval_accuracy | 0.94112 |
|
| 123 |
+
| eval_auc | 0.9849 |
|
| 124 |
+
| eval_f1_score | 0.9417 |
|
| 125 |
+
| eval_precision | 0.9321 |
|
| 126 |
+
| eval_recall | 0.95149 |
|
| 127 |
### Framework versions
|
| 128 |
|
| 129 |
- Transformers 4.34.0
|