Это часть учебного проекта по машинному обучению
Проект английский, поэтому прикрепляю перевод алгоритма действий, которые нужно выполнить
А это ссылка на оригинал:
Так же ссылка на книгу, на основе которой написан алгоритм(от туда взяты названия классов и функций):