michelecafagna26
commited on
Commit
·
edd72d3
1
Parent(s):
2f5b06b
Update README.md
Browse filesUpdated evaluation results and new output format. Fixed previous model
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 |
|
27 |
-
|positive |
|
28 |
|----------|----------|---------|----------|-------|
|
29 |
-
|accuracy| | |
|
30 |
-
|macro avg|
|
31 |
-
|weighted avg|
|
32 |
|
33 |
|
34 |
## Model in Action 🚀
|
@@ -49,7 +49,8 @@ def get_sentiment(text):
|
|
49 |
|
50 |
get_sentiment("This movie is awesome")
|
51 |
|
52 |
-
#
|
|
|
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
|