3oclock commited on
Commit
89f0f5f
·
verified ·
1 Parent(s): b3ced13

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -7
README.md CHANGED
@@ -21,10 +21,13 @@ model-index:
21
  type: accuracy
22
  value: 0.9316
23
  ---
 
 
 
24
 
25
  ## Performance
26
- - Loss: 0.19578103721141815
27
- - Accuracy: 0.9316129032258065
28
 
29
  ## How to Get Started with the Model
30
 
@@ -38,11 +41,6 @@ classifier = pipeline("sentiment-analysis", model="3oclock/distilbert-imdb", tok
38
  result = classifier("I love this movie!")
39
  print(result)
40
  ```
41
-
42
- # Model Card for DistilBERT Fine-Tuned on IMDb Dataset
43
-
44
- This model is a fine-tuned version of `distilbert-base-uncased` on the IMDb movie reviews dataset. It is intended for binary sentiment classification (positive or negative) of movie reviews.
45
-
46
  ## Model Details
47
 
48
  ### Model Description
 
21
  type: accuracy
22
  value: 0.9316
23
  ---
24
+ # distilbert-imdb
25
+
26
+ This is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on imdb dataset.
27
 
28
  ## Performance
29
+ - Loss: 0.1958
30
+ - Accuracy: 0.932
31
 
32
  ## How to Get Started with the Model
33
 
 
41
  result = classifier("I love this movie!")
42
  print(result)
43
  ```
 
 
 
 
 
44
  ## Model Details
45
 
46
  ### Model Description