https://gist.github.com/adarsh1021/b33108e9a6c899583afabf169b6bae6d#file-linear_regression_gd1-py