sklearn linear regression example with categorical variables