Mentis interview question

Take home assignment relating to ML