Kerassy commited on
Commit
193d061
·
verified ·
1 Parent(s): 27e5f82

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -0
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)