linear regression python code with dataset github