michelecafagna26 commited on
Commit
edd72d3
•
1 Parent(s): 2f5b06b

Update README.md

Browse files

Updated evaluation results and new output format. Fixed previous model

Files changed (1) hide show
  1. README.md +7 -6
README.md CHANGED
@@ -23,12 +23,12 @@ The model has been finetuned for 10 epochs on standard hyperparameters
23
 
24
  |precision | recall | f1-score |support|
25
  |----------|----------|---------|----------|-------|
26
- |negative | 1.00 | 1.00| 1.00| 428 |
27
- |positive | 1.00 | 1.00| 1.00| 444 |
28
  |----------|----------|---------|----------|-------|
29
- |accuracy| | | 1.00| 872 |
30
- |macro avg| 1.00| 1.00| 1.00| 872 |
31
- |weighted avg| 1.00| 1.00| 1.00 | 872 |
32
 
33
 
34
  ## Model in Action 🚀
@@ -49,7 +49,8 @@ def get_sentiment(text):
49
 
50
  get_sentiment("This movie is awesome")
51
 
52
- # Output: ['positive']
 
53
  ```
54
 
55
  > This model card is based on "mrm8488/t5-base-finetuned-imdb-sentiment" by Manuel Romero/@mrm8488
 
23
 
24
  |precision | recall | f1-score |support|
25
  |----------|----------|---------|----------|-------|
26
+ |negative | 0.95 | 0.95| 0.95| 428 |
27
+ |positive | 0.94 | 0.96| 0.95| 444 |
28
  |----------|----------|---------|----------|-------|
29
+ |accuracy| | | 0.95| 872 |
30
+ |macro avg| 0.95| 0.95| 0.95| 872 |
31
+ |weighted avg| 0.95| 0.95| 0.95 | 872 |
32
 
33
 
34
  ## Model in Action 🚀
 
49
 
50
  get_sentiment("This movie is awesome")
51
 
52
+ # labels are 'p' for 'positive' and 'n' for 'negative'
53
+ # Output: ['p']
54
  ```
55
 
56
  > This model card is based on "mrm8488/t5-base-finetuned-imdb-sentiment" by Manuel Romero/@mrm8488