bhadresh-savani
/

albert-base-v2-emotion

Text Classification

Model card Files Files and versions

bhadresh-savani commited on Jun 18, 2021

Commit

0cc4406

·

1 Parent(s): 97dcd15

Update README.md

Files changed (1) hide show

README.md +6 -6

README.md CHANGED Viewed

@@ -43,12 +43,12 @@ print(prediction)
 """
 Output:
 [[
-{'label': 'sadness', 'score': 0.0006792712374590337},
-{'label': 'joy', 'score': 0.9959300756454468},
-{'label': 'love', 'score': 0.0009452480007894337},
-{'label': 'anger', 'score': 0.0018055217806249857},
-{'label': 'fear', 'score': 0.00041110432357527316},
-{'label': 'surprise', 'score': 0.0002288572577526793}
 ]]
 """
 ```

 """
 Output:
 [[
+{'label': 'sadness', 'score': 0.010403595864772797},
+{'label': 'joy', 'score': 0.8902180790901184},
+{'label': 'love', 'score': 0.042532723397016525},
+{'label': 'anger', 'score': 0.041297927498817444},
+{'label': 'fear', 'score': 0.011772023513913155},
+{'label': 'surprise', 'score': 0.0037756056990474463}
 ]]
 """
 ```