Final project of the machine learning course