Hugging Face's logo

florianehmann
/
distilbert-base-uncased-finetuned-emotion

Text Classification
Transformers
PyTorch
distilbert
Generated from Trainer
Eval Results
Inference Endpoints
Model card Files Community
florianehmann commited on
Commit
3759aca
•
1 Parent(s): 62028e9

Training completed!

Browse files
Files changed (4) hide show
  1. README.md +8 -8
  2. pytorch_model.bin +1 -1
  3. tokenizer.json +14 -2
  4. training_args.bin +1 -1
README.md CHANGED
@@ -23,10 +23,10 @@ model-index:
23
  metrics:
24
  - name: Accuracy
25
  type: accuracy
26
- value: 0.934
27
  - name: F1
28
  type: f1
29
- value: 0.934088969527908
30
  ---
31
 
32
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -36,9 +36,9 @@ should probably proofread and complete it, then remove this comment. -->
36
 
37
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the emotion dataset.
38
  It achieves the following results on the evaluation set:
39
- - Loss: 0.1524
40
- - Accuracy: 0.934
41
- - F1: 0.9341
42
 
43
  ## Model description
44
 
@@ -69,9 +69,9 @@ The following hyperparameters were used during training:
69
 
70
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
71
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
72
- | 0.1746 | 1.0 | 250 | 0.1814 | 0.931 | 0.9318 |
73
- | 0.1148 | 2.0 | 500 | 0.1630 | 0.935 | 0.9357 |
74
- | 0.097 | 3.0 | 750 | 0.1524 | 0.934 | 0.9341 |
75
 
76
 
77
  ### Framework versions
 
23
  metrics:
24
  - name: Accuracy
25
  type: accuracy
26
+ value: 0.9315
27
  - name: F1
28
  type: f1
29
+ value: 0.9315678224371554
30
  ---
31
 
32
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
36
 
37
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the emotion dataset.
38
  It achieves the following results on the evaluation set:
39
+ - Loss: 0.1703
40
+ - Accuracy: 0.9315
41
+ - F1: 0.9316
42
 
43
  ## Model description
44
 
 
69
 
70
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
71
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
72
+ | 0.8038 | 1.0 | 250 | 0.2937 | 0.9095 | 0.9087 |
73
+ | 0.223 | 2.0 | 500 | 0.1875 | 0.927 | 0.9277 |
74
+ | 0.1549 | 3.0 | 750 | 0.1703 | 0.9315 | 0.9316 |
75
 
76
 
77
  ### Framework versions
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5d120f2dd0b0087744958e1a86dc68a46300f8b45499874e14e89c522f785aaa
3
  size 267868266
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e6fda24e9f05351117ee115ed2ef5b4a66d38bc32561f1b2ac33f25ed47ad4cc
3
  size 267868266
tokenizer.json CHANGED
@@ -1,7 +1,19 @@
1
  {
2
  "version": "1.0",
3
- "truncation": null,
4
- "padding": null,
 
 
 
 
 
 
 
 
 
 
 
 
5
  "added_tokens": [
6
  {
7
  "id": 0,
 
1
  {
2
  "version": "1.0",
3
+ "truncation": {
4
+ "direction": "Right",
5
+ "max_length": 512,
6
+ "strategy": "LongestFirst",
7
+ "stride": 0
8
+ },
9
+ "padding": {
10
+ "strategy": "BatchLongest",
11
+ "direction": "Right",
12
+ "pad_to_multiple_of": null,
13
+ "pad_id": 0,
14
+ "pad_type_id": 0,
15
+ "pad_token": "[PAD]"
16
+ },
17
  "added_tokens": [
18
  {
19
  "id": 0,
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bf4e43b3e9ab99387a351499b162952ea0b36ce0a8a89bb09e8b3cb94d1216f9
3
  size 4536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c04b664ccc30144ab2ac88e90c981b7fee65cabd5452dae86f4dfda379baa2b
3
  size 4536