hamayun7333 commited on
Commit
929d039
1 Parent(s): d43ceb0

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -24
README.md CHANGED
@@ -3,30 +3,12 @@ license: apache-2.0
3
  base_model: distilbert-base-uncased
4
  tags:
5
  - generated_from_trainer
6
- datasets:
7
- - imdb
8
  metrics:
9
  - accuracy
10
  - f1
11
  model-index:
12
  - name: finetuning-sentiment-model-3000-samples
13
- results:
14
- - task:
15
- name: Text Classification
16
- type: text-classification
17
- dataset:
18
- name: imdb
19
- type: imdb
20
- config: plain_text
21
- split: test
22
- args: plain_text
23
- metrics:
24
- - name: Accuracy
25
- type: accuracy
26
- value: 0.8733333333333333
27
- - name: F1
28
- type: f1
29
- value: 0.8741721854304636
30
  ---
31
 
32
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -34,11 +16,11 @@ should probably proofread and complete it, then remove this comment. -->
34
 
35
  # finetuning-sentiment-model-3000-samples
36
 
37
- This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the imdb dataset.
38
  It achieves the following results on the evaluation set:
39
- - Loss: 0.3211
40
- - Accuracy: 0.8733
41
- - F1: 0.8742
42
 
43
  ## Model description
44
 
@@ -73,5 +55,4 @@ The following hyperparameters were used during training:
73
 
74
  - Transformers 4.31.0
75
  - Pytorch 2.0.1+cu118
76
- - Datasets 2.14.0
77
  - Tokenizers 0.13.3
 
3
  base_model: distilbert-base-uncased
4
  tags:
5
  - generated_from_trainer
 
 
6
  metrics:
7
  - accuracy
8
  - f1
9
  model-index:
10
  - name: finetuning-sentiment-model-3000-samples
11
+ results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
  ---
13
 
14
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
16
 
17
  # finetuning-sentiment-model-3000-samples
18
 
19
+ This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.3286
22
+ - Accuracy: 0.87
23
+ - F1: 0.8730
24
 
25
  ## Model description
26
 
 
55
 
56
  - Transformers 4.31.0
57
  - Pytorch 2.0.1+cu118
 
58
  - Tokenizers 0.13.3