Update README.md
Browse files
README.md
CHANGED
|
@@ -27,17 +27,6 @@ widget:
|
|
| 27 |
"other": 5
|
| 28 |
}
|
| 29 |
|
| 30 |
-
## How to Use:
|
| 31 |
-
|
| 32 |
-
Here is an example of how to use this model for inference:
|
| 33 |
-
|
| 34 |
-
```python
|
| 35 |
-
from transformers import pipeline
|
| 36 |
-
|
| 37 |
-
classifier = pipeline("text-classification", model="dnzblgn/Customer-Reviews-Classification")
|
| 38 |
-
result = classifier("The product arrived on time and was exactly as described.")
|
| 39 |
-
print(result)
|
| 40 |
-
|
| 41 |
### Model Description
|
| 42 |
|
| 43 |
This fine-tuned DistilBERT model is specifically designed for document classification. It classifies customer feedback into six predefined categories: Shipping and Delivery, Customer Service, Price and Value, Quality and Performance, Use and Design, and Other. By leveraging the transformer-based architecture of DistilBERT, the model efficiently handles the syntactic patterns of text, providing accurate document classification based on content, style, and structure.
|
|
@@ -87,4 +76,15 @@ Recall: 0.948
|
|
| 87 |
F1-Score: 0.948
|
| 88 |
|
| 89 |
|
| 90 |
-
### For access to the synthetic dataset used, please contact: [deniz.bilgin@uni-konstanz.de].
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
"other": 5
|
| 28 |
}
|
| 29 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
### Model Description
|
| 31 |
|
| 32 |
This fine-tuned DistilBERT model is specifically designed for document classification. It classifies customer feedback into six predefined categories: Shipping and Delivery, Customer Service, Price and Value, Quality and Performance, Use and Design, and Other. By leveraging the transformer-based architecture of DistilBERT, the model efficiently handles the syntactic patterns of text, providing accurate document classification based on content, style, and structure.
|
|
|
|
| 76 |
F1-Score: 0.948
|
| 77 |
|
| 78 |
|
| 79 |
+
### For access to the synthetic dataset used, please contact: [deniz.bilgin@uni-konstanz.de].
|
| 80 |
+
|
| 81 |
+
## How to Use:
|
| 82 |
+
|
| 83 |
+
Here is an example of how to use this model for inference:
|
| 84 |
+
|
| 85 |
+
```python
|
| 86 |
+
from transformers import pipeline
|
| 87 |
+
|
| 88 |
+
classifier = pipeline("text-classification", model="dnzblgn/Customer-Reviews-Classification")
|
| 89 |
+
result = classifier("The product arrived on time and was exactly as described.")
|
| 90 |
+
print(result)
|