AlekseyKorshuk commited on
Commit
474b3e5
·
1 Parent(s): d289c3d

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -21
README.md CHANGED
@@ -2,24 +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: bert
11
- results:
12
- - task:
13
- name: Text Classification
14
- type: text-classification
15
- dataset:
16
- name: imdb
17
- type: imdb
18
- args: plain_text
19
- metrics:
20
- - name: Accuracy
21
- type: accuracy
22
- value: 0.92864
23
  ---
24
 
25
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -27,10 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
27
 
28
  # bert
29
 
30
- This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the imdb dataset.
31
  It achieves the following results on the evaluation set:
32
- - Loss: 0.1934
33
- - Accuracy: 0.9286
34
 
35
  ## Model description
36
 
@@ -63,10 +50,10 @@ The following hyperparameters were used during training:
63
 
64
  ### Training results
65
 
66
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
- |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
- | No log | 1.0 | 391 | 0.2053 | 0.9210 |
69
- | 0.3142 | 2.0 | 782 | 0.1934 | 0.9286 |
70
 
71
 
72
  ### Framework versions
 
2
  license: apache-2.0
3
  tags:
4
  - generated_from_trainer
 
 
5
  metrics:
6
  - accuracy
7
  model-index:
8
  - name: bert
9
+ results: []
 
 
 
 
 
 
 
 
 
 
 
10
  ---
11
 
12
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
14
 
15
  # bert
16
 
17
+ This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 1.5356
20
+ - Accuracy: 0.2900
21
 
22
  ## Model description
23
 
 
50
 
51
  ### Training results
52
 
53
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
+ |:-------------:|:-----:|:-----:|:---------------:|:--------:|
55
+ | 1.5358 | 1.0 | 6195 | 1.5355 | 0.2908 |
56
+ | 1.5248 | 2.0 | 12390 | 1.5356 | 0.2900 |
57
 
58
 
59
  ### Framework versions