/
pash1ka
/
question-difficulty-classifier
Обзор
Документация
Войти
/
pash1ka
/
question-difficulty-classifier
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
src/classifier.py
29 строк
891 B
Blinov Pavel
Final project: Question Difficulty Classifier 15/15 ready
07 дек 2025, 16:09
07 дек 2025, 16:09
c285405
Код
Авторство
О чём код?
import joblib from .features import extract_features import warnings warnings.filterwarnings("ignore", category=UserWarning, module="sklearn") def classify_question( question_text: str, pct_correct: float, avg_time_sec: float, model_path: str = "model.pkl", ): # Загружаем обученную модель model = joblib.load(model_path) # Определяем порядок признаков feature_order = ["length", "word_count", "fk_grade", "pct_correct", "avg_time_sec"] # Извлекаем признаки из текста вопроса features = extract_features(question_text, pct_correct, avg_time_sec) # Формируем вектор признаков X = [[features[k] for k in feature_order]] # Предсказываем сложность prediction = model.predict(X)[0] return prediction