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