def __init__(self): # loss="log" makes it use logistic regression self.model = linear_model.SGDClassifier(loss="log", n_iter=5)