Remove more unneeded code
Switch back to scikit-learn Linear Regression model

This was originally a custom one because the one in scikit-learn
has blown up in the past before for no reason at all.