linear regression python code github