Update README.md
Browse files
README.md
CHANGED
|
@@ -27,4 +27,22 @@ text = "The hotel was amazing and had great service!"
|
|
| 27 |
inputs = tokenizer(text, return_tensors="pt")
|
| 28 |
outputs = model(**inputs)
|
| 29 |
predicted_class = outputs.logits.argmax().item()
|
| 30 |
-
print(f"Predicted class: {predicted_class}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
inputs = tokenizer(text, return_tensors="pt")
|
| 28 |
outputs = model(**inputs)
|
| 29 |
predicted_class = outputs.logits.argmax().item()
|
| 30 |
+
print(f"Predicted class: {predicted_class}")
|
| 31 |
+
```
|
| 32 |
+
|
| 33 |
+
## Testing results
|
| 34 |
+
- **Evaluation accuracy**: 0.6535
|
| 35 |
+
- **Precision**: 0.635
|
| 36 |
+
- **Recall**: 0.641
|
| 37 |
+
- **F1 score**: 0.636
|
| 38 |
+
- **Confusion matrix**:
|
| 39 |
+
```
|
| 40 |
+
| Predicted → | 1 | 2 | 3 | 4 | 5 |
|
| 41 |
+
|---------------|------|------|------|------|------|
|
| 42 |
+
| Actual ↓ | | | | | |
|
| 43 |
+
| 1 (Very Neg.) | 1219 | 318 | 48 | 6 | 9 |
|
| 44 |
+
| 2 (Negative) | 432 | 826 | 294 | 32 | 16 |
|
| 45 |
+
| 3 (Neutral) | 51 | 306 | 928 | 275 | 40 |
|
| 46 |
+
| 4 (Positive) | 3 | 22 | 223 | 833 | 519 |
|
| 47 |
+
| 5 (Very Pos.) | 9 | 6 | 16 | 247 | 1322 |
|
| 48 |
+
```
|