kfahn commited on
Commit
00cceda
1 Parent(s): fe04bd6

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -11
README.md CHANGED
@@ -21,7 +21,7 @@ model-index:
21
  metrics:
22
  - name: Accuracy
23
  type: accuracy
24
- value: 0.8
25
  ---
26
 
27
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -31,8 +31,8 @@ should probably proofread and complete it, then remove this comment. -->
31
 
32
  This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the GTZAN dataset.
33
  It achieves the following results on the evaluation set:
34
- - Loss: 0.6772
35
- - Accuracy: 0.8
36
 
37
  ## Model description
38
 
@@ -64,14 +64,14 @@ The following hyperparameters were used during training:
64
 
65
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
66
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
67
- | 1.8944 | 1.0 | 110 | 1.8362 | 0.5167 |
68
- | 1.4359 | 2.0 | 220 | 1.4329 | 0.5083 |
69
- | 0.9861 | 3.0 | 330 | 1.0460 | 0.7 |
70
- | 0.9073 | 4.0 | 440 | 0.8689 | 0.7417 |
71
- | 0.5268 | 5.0 | 550 | 0.8289 | 0.8 |
72
- | 0.4683 | 6.0 | 660 | 0.7483 | 0.7833 |
73
- | 0.2342 | 7.0 | 770 | 0.7025 | 0.8 |
74
- | 0.2886 | 8.0 | 880 | 0.6772 | 0.8 |
75
 
76
 
77
  ### Framework versions
 
21
  metrics:
22
  - name: Accuracy
23
  type: accuracy
24
+ value: 0.85
25
  ---
26
 
27
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
31
 
32
  This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the GTZAN dataset.
33
  It achieves the following results on the evaluation set:
34
+ - Loss: 0.5486
35
+ - Accuracy: 0.85
36
 
37
  ## Model description
38
 
 
64
 
65
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
66
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
67
+ | 1.9391 | 1.0 | 113 | 1.7463 | 0.42 |
68
+ | 1.3438 | 2.0 | 226 | 1.1953 | 0.64 |
69
+ | 1.04 | 3.0 | 339 | 0.9436 | 0.72 |
70
+ | 0.8608 | 4.0 | 452 | 0.8464 | 0.74 |
71
+ | 0.5644 | 5.0 | 565 | 0.6846 | 0.8 |
72
+ | 0.4311 | 6.0 | 678 | 0.7118 | 0.77 |
73
+ | 0.509 | 7.0 | 791 | 0.5660 | 0.83 |
74
+ | 0.2548 | 8.0 | 904 | 0.5486 | 0.85 |
75
 
76
 
77
  ### Framework versions