Esempio n. 1
0
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])