Accuracy Report
Browse files
README.md
CHANGED
|
@@ -36,7 +36,38 @@ Instead of fine-tuning all parameters, **Low-Rank Adaptation (LoRA)** was applie
|
|
| 36 |
- **Dataset:** https://huggingface.co/datasets/imdb
|
| 37 |
- **Repository:** https://huggingface.co/ChetanFernandis/distilbert_cls-lora-IMDB
|
| 38 |
|
| 39 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
|
| 41 |
## Uses
|
| 42 |
|
|
|
|
| 36 |
- **Dataset:** https://huggingface.co/datasets/imdb
|
| 37 |
- **Repository:** https://huggingface.co/ChetanFernandis/distilbert_cls-lora-IMDB
|
| 38 |
|
| 39 |
+
## Evaluation Results
|
| 40 |
+
|
| 41 |
+
The model was evaluated on a held-out validation subset of the **IMDB dataset** using standard classification metrics.
|
| 42 |
+
|
| 43 |
+
### Confusion Matrix
|
| 44 |
+
|
| 45 |
+
- **NEGATIVE → NEGATIVE:** 42
|
| 46 |
+
- **NEGATIVE → POSITIVE:** 11
|
| 47 |
+
- **POSITIVE → NEGATIVE:** 7
|
| 48 |
+
- **POSITIVE → POSITIVE:** 40
|
| 49 |
+
|
| 50 |
+
This indicates balanced performance across both sentiment classes.
|
| 51 |
+
|
| 52 |
+
|
| 53 |
+
### Classification Report
|
| 54 |
+
|
| 55 |
+
precision recall f1-score support
|
| 56 |
+
|
| 57 |
+
NEGATIVE 0.86 0.79 0.82 53
|
| 58 |
+
POSITIVE 0.78 0.85 0.82 47
|
| 59 |
+
|
| 60 |
+
accuracy 0.82 100
|
| 61 |
+
|
| 62 |
+
### Summary
|
| 63 |
+
|
| 64 |
+
- **Overall Accuracy:** 82%
|
| 65 |
+
- **Balanced F1-score:** 0.82 for both classes
|
| 66 |
+
- Strong precision for **NEGATIVE** reviews
|
| 67 |
+
- Strong recall for **POSITIVE** reviews
|
| 68 |
+
|
| 69 |
+
These results demonstrate that **LoRA fine-tuning** achieves competitive sentiment classification performance while training only a small fraction of model parameters.
|
| 70 |
+
|
| 71 |
|
| 72 |
## Uses
|
| 73 |
|