def logistic_regression(dftrain, dftest, predict_col): """ Logistic Regression for HW2 part B""" features = dftrain.columns.tolist() features.remove(predict_col) cls = LogisticRegression() cls.fit(dftrain[features], dftrain[predict_col]) predictions = cls.predict(dftest[features]) print predictions print mystats.compute_ACC(predictions, dftest[predict_col])