Prathamesh2403 commited on
Commit
229e26c
·
verified ·
1 Parent(s): 396461b

Update README.md

Browse files

Output:
"fantastic", "great", "terrible", "fabulous", "perfect"

Files changed (1) hide show
  1. README.md +37 -31
README.md CHANGED
@@ -1,48 +1,54 @@
1
  ---
2
- library_name: transformers
 
3
  tags:
4
- - generated_from_keras_callback
5
- model-index:
6
- - name: distilbert-base-uncased-finetuned-imdb
7
- results: []
 
 
 
 
 
8
  ---
9
 
10
- <!-- This model card has been generated automatically according to the information Keras had access to. You should
11
- probably proofread and complete it, then remove this comment. -->
12
-
13
  # distilbert-base-uncased-finetuned-imdb
14
 
15
- This model was trained from scratch on an unknown dataset.
16
- It achieves the following results on the evaluation set:
17
-
18
-
19
- ## Model description
20
-
21
- More information needed
22
 
23
- ## Intended uses & limitations
24
-
25
- More information needed
26
 
27
- ## Training and evaluation data
28
 
29
- More information needed
 
 
 
 
 
30
 
31
- ## Training procedure
32
 
33
- ### Training hyperparameters
34
 
35
- The following hyperparameters were used during training:
36
- - optimizer: None
37
- - training_precision: float32
 
38
 
39
- ### Training results
40
 
 
41
 
 
 
42
 
43
- ### Framework versions
 
 
 
 
44
 
45
- - Transformers 4.52.4
46
- - TensorFlow 2.18.0
47
- - Datasets 3.6.0
48
- - Tokenizers 0.21.1
 
1
  ---
2
+ language: en
3
+ license: apache-2.0
4
  tags:
5
+ - masked-language-modeling
6
+ - imdb
7
+ - distilbert
8
+ - domain-adaptation
9
+ - tensorflow
10
+ - movie-reviews
11
+ pipeline_tag: fill-mask
12
+ widget:
13
+ - text: "This movie was absolutely [MASK]."
14
  ---
15
 
 
 
 
16
  # distilbert-base-uncased-finetuned-imdb
17
 
18
+ This is a **DistilBERT** model fine-tuned using **Masked Language Modeling (MLM)** on the **IMDB movie reviews dataset**.
19
+ It is domain-adapted specifically for understanding and completing movie-related text.
 
 
 
 
 
20
 
21
+ ---
 
 
22
 
23
+ ## Model Details
24
 
25
+ - **Base model**: [`distilbert-base-uncased`](https://huggingface.co/distilbert-base-uncased)
26
+ - **Training objective**: Masked Language Modeling
27
+ - **Domain**: English movie reviews (IMDB)
28
+ - **Framework**: TensorFlow / Keras
29
+ - **Training time**: ~3 hours on Google Colab
30
+ - **Chunk size**: 128 tokens
31
 
32
+ ---
33
 
34
+ ## Use Case
35
 
36
+ This model is ideal for:
37
+ - Autocompletion of masked tokens in movie reviews
38
+ - Domain-aware masked language modeling
39
+ - Sentence generation or augmentation in film-related contexts
40
 
41
+ ---
42
 
43
+ ## Example Usage
44
 
45
+ ```python
46
+ from transformers import pipeline
47
 
48
+ fill_mask = pipeline(
49
+ "fill-mask",
50
+ model="Prathamesh2403/distilbert-base-uncased-finetuned-imdb",
51
+ tokenizer="Prathamesh2403/distilbert-base-uncased-finetuned-imdb"
52
+ )
53
 
54
+ fill_mask("This movie was absolutely [MASK].")