https://progiez.com/nptel-introduction-to-machine-learning-assignment-2