Update README.md
Browse files
README.md
CHANGED
|
@@ -45,6 +45,7 @@ with torch.no_grad():
|
|
| 45 |
out = model(**inp).logits.squeeze(0).tolist()
|
| 46 |
OCEAN = ["openness","conscientiousness","extraversion","agreeableness","neuroticism"]
|
| 47 |
print(dict(zip(OCEAN, out)))
|
|
|
|
| 48 |
|
| 49 |
# mini-batch
|
| 50 |
def predict_personality(texts, batch_size=32, max_len=512):
|
|
|
|
| 45 |
out = model(**inp).logits.squeeze(0).tolist()
|
| 46 |
OCEAN = ["openness","conscientiousness","extraversion","agreeableness","neuroticism"]
|
| 47 |
print(dict(zip(OCEAN, out)))
|
| 48 |
+
# OCEAN: {'openness': 0.6432028412818909, 'conscientiousness': 0.7445886135101318, 'extraversion': 0.20433923602104187, 'agreeableness': 0.4930797815322876, 'neuroticism': 0.33562132716178894}
|
| 49 |
|
| 50 |
# mini-batch
|
| 51 |
def predict_personality(texts, batch_size=32, max_len=512):
|