Для репликации обучения нужно положить .csv-файлы в директорию data/
В коде для обучения модели некоторые выводы оформлены в [yellow] жёлтом [/yellow] тексте. У меня стоит тема Gruvbox Dark, поэтому это выглядит красиво. Если вам кажется это неприятным глазу, поменяйте [yellow] на любой другой цвет или вовсе его удалите