EGE Classification

Этот проект предназначен для классификации задач из профильного ЕГЭ по математике. Модель предсказывает номер задачи с точностью 98.6%.

Установка

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

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.