Harsha901 commited on
Commit
68324aa
·
verified ·
1 Parent(s): 6ffd898

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +33 -1
README.md CHANGED
@@ -13,4 +13,36 @@ tags:
13
  - imdb
14
  - fine-tuned
15
  - text-classification
16
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  - imdb
14
  - fine-tuned
15
  - text-classification
16
+ ---
17
+
18
+ # 📊 Sentiment Analysis with Fine-Tuned BERT (IMDB)
19
+
20
+ This repository contains a fine-tuned BERT model for binary sentiment classification using the IMDB movie reviews dataset. The model classifies reviews as **positive** or **negative**, and is built using Hugging Face Transformers and PyTorch.
21
+
22
+ ## 🚀 Model Performance
23
+
24
+ | Metric | Value |
25
+ |------------------|-------------|
26
+ | Accuracy | 89.4% |
27
+ | Validation Loss | 0.375 |
28
+ | Epochs Trained | 3 |
29
+ | Inference Speed | ~434 samples/sec |
30
+
31
+ ## 🧠 Model Details
32
+
33
+ - **Base Model**: `bert-base-uncased`
34
+ - **Dataset**: IMDB (binary sentiment)
35
+ - **Framework**: Hugging Face Transformers
36
+ - **Fine-Tuning Setup**:
37
+ - Learning rate: 2e-5
38
+ - Batch size: 32
39
+ - Mixed-precision: ✅ (`fp16`)
40
+ - Early stopping: ❌ (trained for full 3 epochs)
41
+
42
+ ## 🛠️ How to Use
43
+
44
+ ```python
45
+ from transformers import pipeline
46
+
47
+ classifier = pipeline("text-classification", model="Harsha901/tinybert-imdb-sentiment-analysis-model")
48
+ classifier("This movie was absolutely amazing!")