train.ipynb - ноутбук с кодом для обучения модели
inference.ipynb - ноутбук с кодом инференса и расчетом метрики Accuracy на тестовой выборке
В папке models лежит обученная модель
Перед обучением сбалансировал данные, удалил "дубликаты".
Для обучения модели и инференса необходимо скачать данные и распаковать в директорию. Затем запустить train.ipynb
Для инференса и расчета Accuracy необходимо запустить inference.ipynb
На тестовой выборке удалось достичь Accuracy = 0.72