https://medium.com/better-programming/how-to-implement-linear-regression-in-scikit-learn-e459f6f6a3eb