Commit ·
b876773
1
Parent(s): c5c8dd0
Update README.md
Browse files
README.md
CHANGED
|
@@ -48,6 +48,21 @@ The code above will output following list:
|
|
| 48 |
["negative","negative","positive","positive","neutral", "neutral"]
|
| 49 |
```
|
| 50 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
|
| 52 |
## Model and Data
|
| 53 |
|
|
|
|
| 48 |
["negative","negative","positive","positive","neutral", "neutral"]
|
| 49 |
```
|
| 50 |
|
| 51 |
+
### Output class probabilities
|
| 52 |
+
|
| 53 |
+
```python
|
| 54 |
+
from germansentiment import SentimentModel
|
| 55 |
+
|
| 56 |
+
model = SentimentModel()
|
| 57 |
+
|
| 58 |
+
classes, probabilities = model.predict_sentiment(["das ist super"], output_probabilities = True)
|
| 59 |
+
print(classes, probabilities)
|
| 60 |
+
```
|
| 61 |
+
```python
|
| 62 |
+
['positive'] [[['positive', 0.9761366844177246], ['negative', 0.023540444672107697], ['neutral', 0.00032294404809363186]]]
|
| 63 |
+
```
|
| 64 |
+
|
| 65 |
+
|
| 66 |
|
| 67 |
## Model and Data
|
| 68 |
|