dorukan commited on
Commit
c88900a
·
verified ·
1 Parent(s): 0f369aa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +50 -3
README.md CHANGED
@@ -1,3 +1,50 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ language:
4
+ - en
5
+ base_model:
6
+ - distilbert/distilbert-base-uncased
7
+ ---
8
+ # DistilBERT Sentiment Analysis Model
9
+
10
+ This model is a fine-tuned version of **DistilBERT** for sentiment analysis on the **IMDb** dataset. It classifies movie reviews as either **positive** or **negative** based on the text content.
11
+
12
+ ## Model Details
13
+
14
+ - **Model Type**: DistilBERT (a smaller and faster variant of BERT)
15
+ - **Task**: Sentiment Analysis
16
+ - **Dataset**: IMDb dataset containing movie reviews with labels (positive/negative)
17
+ - **Fine-Tuned On**: IMDb dataset
18
+
19
+ ## Model Performance
20
+
21
+ This model was fine-tuned on the IMDb dataset for sentiment classification, achieving good performance for binary sentiment classification tasks (positive/negative).
22
+
23
+ ## Usage
24
+
25
+ To use this model, you can load it from the Hugging Face Model Hub using the `transformers` library:
26
+
27
+ ```python
28
+ from transformers import pipeline
29
+
30
+ # Load the model
31
+ classifier = pipeline('sentiment-analysis', model='dorukan/distil-bert-imdb')
32
+
33
+ # Example usage
34
+ result = classifier("This movie was amazing!")
35
+ print(result)
36
+ ```
37
+
38
+ This will output a sentiment prediction for the given text.
39
+
40
+
41
+ ## License
42
+
43
+ This model is licensed under the MIT License. For more information, see the LICENSE file.
44
+
45
+ ## Acknowledgments
46
+
47
+ - **DistilBERT**: A smaller version of BERT, created by the Hugging Face team.
48
+ - **IMDb Dataset**: A collection of movie reviews used for sentiment classification, widely used in NLP tasks.
49
+
50
+ You can find more details about the model at the [Hugging Face model page](https://huggingface.co/dorukan/distil-bert-imdb).