https://medium.com/@jw1809131/msc-applied-machine-learning-assignment-homework-help-imperial-college-london-uk-f3b16f794d2e