Using LogisticRegression instead of LinearRegression.