MatveyMerzlikin/ege-classification

GitHub: https://github.com/Merzlikin-Matvey/EGE-classification

EGE Classification

Этот проект предназначен для классификации задач из профильного ЕГЭ по математике. Модель предсказывает номер задачи с точностью более 99%. Главная проблема - разделение 4 и 5 задач (теория вероятностей). Актуальная версия обучена на базе cointegrated/rubert-tiny2 (около 20M параметров). Старая версия (0.0.4 модуля; v1 модели на hf) обучена на базе distilbert-base-uncased (около 70M параметров). Настоятельно рекомендую актуальную версию. Задачи актуальны на Июль 2026 года.

Установка

Вы можете самостоятельно собрать библиотеку

poetry build
pip install ПУТЬ_К_АРХИВУ

Или воспользоваться pypi.org

pip install ege-classification

Использование

Обучение модели

Для обучения модели используйте следующий код:

from ege_classification.model import TaskClassifier

classifier = TaskClassifier()
classifier.train('path/to/dataset.csv')
classifier.save()

Тестирование модели

Для тестирования модели используйте следующий код:

classifier.test('path/to/dataset.csv')

Предсказание

Для предсказания номера задачи используйте следующий код:

predicted_class = classifier.predict("Текст задачи")
print(f"Предсказанный номер задачи: {predicted_class}")

Загрузка и сохранение модели

Для сохранения модели:

classifier.save('path/to/save/model')

Для загрузки модели:

classifier.load('path/to/saved/model')

Лицензия

Этот проект лицензирован под лицензией MIT.