annamalai-s commited on
Commit
409c8e0
·
1 Parent(s): 72838e9

End of training

Browse files
Files changed (1) hide show
  1. README.md +3 -28
README.md CHANGED
@@ -3,30 +3,9 @@ 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: distilbert-sentimentanalysis
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.8566666666666667
27
- - name: F1
28
- type: f1
29
- value: 0.8590163934426229
30
  ---
31
 
32
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -34,11 +13,7 @@ should probably proofread and complete it, then remove this comment. -->
34
 
35
  # distilbert-sentimentanalysis
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.3475
40
- - Accuracy: 0.8567
41
- - F1: 0.8590
42
 
43
  ## Model description
44
 
@@ -73,5 +48,5 @@ The following hyperparameters were used during training:
73
 
74
  - Transformers 4.35.2
75
  - Pytorch 2.1.0+cu118
76
- - Datasets 2.14.7
77
  - Tokenizers 0.15.0
 
3
  base_model: distilbert-base-uncased
4
  tags:
5
  - generated_from_trainer
 
 
 
 
 
6
  model-index:
7
  - name: distilbert-sentimentanalysis
8
+ results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
  ---
10
 
11
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
13
 
14
  # distilbert-sentimentanalysis
15
 
16
+ This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
 
 
 
 
17
 
18
  ## Model description
19
 
 
48
 
49
  - Transformers 4.35.2
50
  - Pytorch 2.1.0+cu118
51
+ - Datasets 2.15.0
52
  - Tokenizers 0.15.0