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