/
NeuroEdTech
/
AI-marafon
Обзор
Документация
Войти
/
NeuroEdTech
/
AI-marafon
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
main
src/inference.py
34 строки
1 KB
Oleg
Initial commit for AI-marafon NLP project
25 янв 2026, 00:18
25 янв 2026, 00:18
b74baaa
Код
Авторство
О чём код?
import joblib from typing import List from .config import MODEL_PATH from .data_prep import basic_clean def load_model(): return joblib.load(MODEL_PATH) def predict_one(text: str) -> str: """ Предсказать класс для одного текста. """ model = load_model() cleaned = basic_clean(text) pred = model.predict([cleaned])[0] return str(pred) def predict_batch(texts: List[str]) -> List[str]: """ Предсказать классы для списка текстов. """ model = load_model() cleaned = [basic_clean(t) for t in texts] preds = model.predict(cleaned) return [str(p) for p in preds] if __name__ == "__main__": example_texts = [ "Модель работает просто отлично, очень доволен результатом!", "Интерфейс неудобный, постоянно всё зависает." ] preds = predict_batch(example_texts) for t, p in zip(example_texts, preds): print(f"Текст: {t}\nКласс: {p}\n")