rebego commited on
Commit
023d1b7
·
verified ·
1 Parent(s): b37e2e4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -5
README.md CHANGED
@@ -12,9 +12,6 @@ model-index:
12
  results: []
13
  ---
14
 
15
- <!-- This model card has been generated automatically according to the information the Trainer had access to. You
16
- should probably proofread and complete it, then remove this comment. -->
17
-
18
  # imdb-distilbert
19
 
20
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
@@ -24,7 +21,10 @@ It achieves the following results on the evaluation set:
24
 
25
  ## Model description
26
 
27
- More information needed
 
 
 
28
 
29
  ## Intended uses & limitations
30
 
@@ -32,7 +32,11 @@ More information needed
32
 
33
  ## Training and evaluation data
34
 
35
- More information needed
 
 
 
 
36
 
37
  ## Training procedure
38
 
 
12
  results: []
13
  ---
14
 
 
 
 
15
  # imdb-distilbert
16
 
17
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 
21
 
22
  ## Model description
23
 
24
+ This model, named imdb-distilbert, is fine-tuned from the distilbert-base-uncased checkpoint on the IMDB movie review dataset for the sentiment classification task.
25
+ It's designed to predict whether a movie review is positive or negative based on the textual content of the review.
26
+ This model can be used to automatically classify new movie reviews into positive or negative categories.
27
+
28
 
29
  ## Intended uses & limitations
30
 
 
32
 
33
  ## Training and evaluation data
34
 
35
+ The model was trained on the IMDB dataset, which contains 50,000 movie reviews split evenly into 25,000 training and 25,000 testing datasets.
36
+ Each entry is labeled as either 0 (negative) or 1 (positive).
37
+ It achieves the following results on the evaluation set:
38
+ - Loss: 1.4670
39
+ - Accuracy: 0.8528
40
 
41
  ## Training procedure
42