linear regression python code with dataset using python