linear regression code in python using sklearn