Разработка алгоритма машинного обучения для стартапа. Идея такая:
- Пациент проходит психологический тест. За ним во время теста наблюдает специальное устройство - айтрекер, получающее из данных с камеры направление взгляда человека.
- На основе полученных данных заранее обученная модель классифицирует человека как скорее здорового или как имеющему склонность к проблемам ментального здоровья (депрессивность, тревожность, ОКР
Датасет на момент моего присоединения к команде уже собрали, моей задачей было сделать на его основе модель. Несмотря на довольно небольшой размер датасета, модели (случайный лес) удалось добиться показателя точности в 75%. По итогу работы стартап успешно прошёл в финал бизнес-акселератора.
