theojolliffe commited on
Commit
6bec817
·
1 Parent(s): 7c378a6

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -21
README.md CHANGED
@@ -2,26 +2,11 @@
2
  license: apache-2.0
3
  tags:
4
  - generated_from_trainer
5
- datasets:
6
- - imdb
7
  metrics:
8
  - accuracy
9
  model-index:
10
  - name: my_awesome_model
11
- results:
12
- - task:
13
- name: Text Classification
14
- type: text-classification
15
- dataset:
16
- name: imdb
17
- type: imdb
18
- config: plain_text
19
- split: train
20
- args: plain_text
21
- metrics:
22
- - name: Accuracy
23
- type: accuracy
24
- value: 0.736
25
  ---
26
 
27
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -29,10 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
29
 
30
  # my_awesome_model
31
 
32
- This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the imdb dataset.
33
  It achieves the following results on the evaluation set:
34
- - Loss: 0.6610
35
- - Accuracy: 0.736
36
 
37
  ## Model description
38
 
@@ -63,8 +48,8 @@ The following hyperparameters were used during training:
63
 
64
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
65
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
66
- | No log | 1.0 | 16 | 0.6785 | 0.616 |
67
- | No log | 2.0 | 32 | 0.6610 | 0.736 |
68
 
69
 
70
  ### Framework versions
 
2
  license: apache-2.0
3
  tags:
4
  - generated_from_trainer
 
 
5
  metrics:
6
  - accuracy
7
  model-index:
8
  - name: my_awesome_model
9
+ results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  ---
11
 
12
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
14
 
15
  # my_awesome_model
16
 
17
+ This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 2.0568
20
+ - Accuracy: 0.3929
21
 
22
  ## Model description
23
 
 
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
+ | No log | 1.0 | 47 | 2.2071 | 0.3333 |
52
+ | No log | 2.0 | 94 | 2.0568 | 0.3929 |
53
 
54
 
55
  ### Framework versions