Update README.md
Browse files
README.md
CHANGED
|
@@ -36,6 +36,10 @@ pipe = pipeline(task="sentiment-analysis", model="Kerassy/bert_base_tp_123k", de
|
|
| 36 |
preds = pipe(reviews)
|
| 37 |
print(preds)
|
| 38 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
### Training Data
|
| 40 |
The model was fine-tuned using the following dataset:
|
| 41 |
[Kerassy/trustpilot-reviews-123k](https://huggingface.co/datasets/Kerassy/trustpilot-reviews-123k)
|
|
|
|
| 36 |
preds = pipe(reviews)
|
| 37 |
print(preds)
|
| 38 |
```
|
| 39 |
+
### Categories
|
| 40 |
+
- 'LABEL_O' - Negative
|
| 41 |
+
- 'LABEL_1' - Neutral
|
| 42 |
+
- 'LABEL_2' - Positive
|
| 43 |
### Training Data
|
| 44 |
The model was fine-tuned using the following dataset:
|
| 45 |
[Kerassy/trustpilot-reviews-123k](https://huggingface.co/datasets/Kerassy/trustpilot-reviews-123k)
|