erlend123 commited on
Commit
598a5eb
1 Parent(s): df75a4e

Model save

Browse files
Files changed (1) hide show
  1. README.md +16 -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,15 @@ 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.3205
22
+ - Accuracy: 0.9015
23
+ - F1: 0.9014
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 | 233 | 0.2691 | 0.9070 | 0.9068 |
55
+ | No log | 2.0 | 466 | 0.2963 | 0.8928 | 0.8922 |
56
+ | 0.2332 | 3.0 | 699 | 0.3205 | 0.9015 | 0.9014 |
57
+
58
+
59
  ### Framework versions
60
 
61
  - Transformers 4.39.3