Update README.md
Browse files
README.md
CHANGED
|
@@ -15,6 +15,8 @@ library_name: transformers
|
|
| 15 |
safetensors: true
|
| 16 |
---
|
| 17 |
|
|
|
|
|
|
|
| 18 |
### Training Details
|
| 19 |
The model was trained on the McAuley-Lab/Amazon-Reviews-2023 dataset. This dataset contains labeled customer reviews from Amazon, focusing on two primary categories: Positive and Negative.
|
| 20 |
|
|
@@ -37,4 +39,11 @@ Precision: 0.98
|
|
| 37 |
|
| 38 |
Recall: 0.99
|
| 39 |
|
| 40 |
-
F1-Score: 0.98
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
safetensors: true
|
| 16 |
---
|
| 17 |
|
| 18 |
+
### sentiment_mapping = {1: "Negative", 0: "Positive"}
|
| 19 |
+
|
| 20 |
### Training Details
|
| 21 |
The model was trained on the McAuley-Lab/Amazon-Reviews-2023 dataset. This dataset contains labeled customer reviews from Amazon, focusing on two primary categories: Positive and Negative.
|
| 22 |
|
|
|
|
| 39 |
|
| 40 |
Recall: 0.99
|
| 41 |
|
| 42 |
+
F1-Score: 0.98
|
| 43 |
+
|
| 44 |
+
```python
|
| 45 |
+
from transformers import pipeline
|
| 46 |
+
|
| 47 |
+
classifier = pipeline("text-classification", model="dnzblgn/Sentiment-Analysis-Customer-Reviews")
|
| 48 |
+
result = classifier("The product didn't arrive on time and was damaged.")
|
| 49 |
+
print(result)
|