gArthur98 commited on
Commit
999d128
·
1 Parent(s): e122c81

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -3
README.md CHANGED
@@ -1,7 +1,9 @@
1
  ---
2
- license: apache-2.0
3
  tags:
4
  - generated_from_trainer
 
 
5
  model-index:
6
  - name: Greg-Sentiment-classifier
7
  results: []
@@ -12,7 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
12
 
13
  # Greg-Sentiment-classifier
14
 
15
- This model is a fine-tuned version of [distilbert-base-cased-distilled-squad](https://huggingface.co/distilbert-base-cased-distilled-squad) on an unknown dataset.
 
 
 
16
 
17
  ## Model description
18
 
@@ -31,7 +36,7 @@ More information needed
31
  ### Training hyperparameters
32
 
33
  The following hyperparameters were used during training:
34
- - learning_rate: 2e-05
35
  - train_batch_size: 16
36
  - eval_batch_size: 16
37
  - seed: 42
@@ -39,6 +44,18 @@ The following hyperparameters were used during training:
39
  - lr_scheduler_type: linear
40
  - num_epochs: 6
41
 
 
 
 
 
 
 
 
 
 
 
 
 
42
  ### Framework versions
43
 
44
  - Transformers 4.30.2
 
1
  ---
2
+ license: mit
3
  tags:
4
  - generated_from_trainer
5
+ metrics:
6
+ - f1
7
  model-index:
8
  - name: Greg-Sentiment-classifier
9
  results: []
 
14
 
15
  # Greg-Sentiment-classifier
16
 
17
+ This model is a fine-tuned version of [microsoft/MiniLM-L12-H384-uncased](https://huggingface.co/microsoft/MiniLM-L12-H384-uncased) on an unknown dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 1.0165
20
+ - F1: 0.3222
21
 
22
  ## Model description
23
 
 
36
  ### Training hyperparameters
37
 
38
  The following hyperparameters were used during training:
39
+ - learning_rate: 0.004
40
  - train_batch_size: 16
41
  - eval_batch_size: 16
42
  - seed: 42
 
44
  - lr_scheduler_type: linear
45
  - num_epochs: 6
46
 
47
+ ### Training results
48
+
49
+ | Training Loss | Epoch | Step | Validation Loss | F1 |
50
+ |:-------------:|:-----:|:----:|:---------------:|:------:|
51
+ | 1.0386 | 1.0 | 499 | 1.0199 | 0.2347 |
52
+ | 1.0437 | 2.0 | 998 | 1.0226 | 0.2347 |
53
+ | 1.066 | 3.0 | 1497 | 1.0207 | 0.2347 |
54
+ | 1.0183 | 4.0 | 1996 | 1.0173 | 0.3222 |
55
+ | 0.9965 | 5.0 | 2495 | 1.0194 | 0.2347 |
56
+ | 0.9548 | 6.0 | 2994 | 1.0165 | 0.3222 |
57
+
58
+
59
  ### Framework versions
60
 
61
  - Transformers 4.30.2