chunwoolee0 commited on
Commit
69c8c98
1 Parent(s): 44a5303

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -36
README.md CHANGED
@@ -1,31 +1,11 @@
1
  ---
2
- license: apache-2.0
3
  tags:
4
  - generated_from_trainer
5
  datasets:
6
  - emotion
7
- metrics:
8
- - accuracy
9
- - f1
10
  model-index:
11
  - name: distilbert-base-uncased-finetuned-emotion
12
- results:
13
- - task:
14
- name: Text Classification
15
- type: text-classification
16
- dataset:
17
- name: emotion
18
- type: emotion
19
- config: split
20
- split: validation
21
- args: split
22
- metrics:
23
- - name: Accuracy
24
- type: accuracy
25
- value: 0.925
26
- - name: F1
27
- type: f1
28
- value: 0.9251030465068828
29
  ---
30
 
31
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -33,11 +13,7 @@ should probably proofread and complete it, then remove this comment. -->
33
 
34
  # distilbert-base-uncased-finetuned-emotion
35
 
36
- This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the emotion dataset.
37
- It achieves the following results on the evaluation set:
38
- - Loss: 0.2205
39
- - Accuracy: 0.925
40
- - F1: 0.9251
41
 
42
  ## Model description
43
 
@@ -64,17 +40,9 @@ The following hyperparameters were used during training:
64
  - lr_scheduler_type: linear
65
  - num_epochs: 2
66
 
67
- ### Training results
68
-
69
- | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
70
- |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
71
- | 0.828 | 1.0 | 250 | 0.3206 | 0.907 | 0.9032 |
72
- | 0.2516 | 2.0 | 500 | 0.2205 | 0.925 | 0.9251 |
73
-
74
-
75
  ### Framework versions
76
 
77
- - Transformers 4.30.2
78
- - Pytorch 2.0.1+cu118
79
  - Datasets 2.13.0
80
  - Tokenizers 0.13.3
 
1
  ---
 
2
  tags:
3
  - generated_from_trainer
4
  datasets:
5
  - emotion
 
 
 
6
  model-index:
7
  - name: distilbert-base-uncased-finetuned-emotion
8
+ results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  ---
10
 
11
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
13
 
14
  # distilbert-base-uncased-finetuned-emotion
15
 
16
+ This model was trained from scratch on the emotion dataset.
 
 
 
 
17
 
18
  ## Model description
19
 
 
40
  - lr_scheduler_type: linear
41
  - num_epochs: 2
42
 
 
 
 
 
 
 
 
 
43
  ### Framework versions
44
 
45
+ - Transformers 4.28.1
46
+ - Pytorch 2.0.1+cu117
47
  - Datasets 2.13.0
48
  - Tokenizers 0.13.3