Amazon interview question

Complete a homework exercise and solve an ML problem.