https://dmachinesb.blogspot.com/2022/06/machine-learning-programming-exercise-5.html