linear regression example using python