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)
|