linear regression model in machine learning code