erlend123 commited on
Commit
8e23ca7
1 Parent(s): 412daca

Model save

Browse files
Files changed (1) hide show
  1. README.md +17 -0
README.md CHANGED
@@ -3,6 +3,9 @@ license: apache-2.0
3
  base_model: distilbert-base-uncased
4
  tags:
5
  - generated_from_trainer
 
 
 
6
  model-index:
7
  - name: emotion-analysis-3000
8
  results: []
@@ -14,6 +17,10 @@ should probably proofread and complete it, then remove this comment. -->
14
  # emotion-analysis-3000
15
 
16
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 
 
 
 
17
 
18
  ## Model description
19
 
@@ -40,6 +47,16 @@ The following hyperparameters were used during training:
40
  - lr_scheduler_type: linear
41
  - num_epochs: 5
42
 
 
 
 
 
 
 
 
 
 
 
43
  ### Framework versions
44
 
45
  - Transformers 4.39.3
 
3
  base_model: distilbert-base-uncased
4
  tags:
5
  - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
+ - f1
9
  model-index:
10
  - name: emotion-analysis-3000
11
  results: []
 
17
  # emotion-analysis-3000
18
 
19
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.3266
22
+ - Accuracy: 0.8949
23
+ - F1: 0.8947
24
 
25
  ## Model description
26
 
 
47
  - lr_scheduler_type: linear
48
  - num_epochs: 5
49
 
50
+ ### Training results
51
+
52
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
53
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
54
+ | No log | 1.0 | 232 | 0.2877 | 0.8938 | 0.8939 |
55
+ | No log | 2.0 | 464 | 0.2719 | 0.9091 | 0.9088 |
56
+ | 0.2784 | 3.0 | 696 | 0.3090 | 0.9003 | 0.9003 |
57
+ | 0.2784 | 4.0 | 928 | 0.3266 | 0.8949 | 0.8947 |
58
+
59
+
60
  ### Framework versions
61
 
62
  - Transformers 4.39.3