Use logistic regression from sklearn to test