/
allelleo
/
UCarTopDoerTestTask
Обзор
Документация
Войти
/
allelleo
/
UCarTopDoerTestTask
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
main
sentiment.py
16 строк
525 B
Овчинников Алексей Александрович
init commit
15 июл 2025, 12:49
15 июл 2025, 12:49
82148b7
Код
Авторство
О чём код?
import torch from settings import tokenizer, model, label_to_sentiment def detect_sentiment(text: str) -> str: inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512) with torch.no_grad(): outputs = model(**inputs) logits = outputs.logits probs = torch.nn.functional.softmax(logits, dim=-1) label_id = torch.argmax(probs).item() # Получаем метку label = model.config.id2label[label_id] return label_to_sentiment.get(label, "neutral")