Linear Regression Github Python