log_reg.py 文件源码

def sm_logit(self,Xtrain,ytrain, Xtest, ytest):
            sm_results = sm.Logit(ytrain, Xtrain).fit_regularized(alpha = 10, disp = False)
            print sm_results.summary()
            # predict train labels
            train_predictions = sm_results.predict(Xtrain)
            train_accuracy = self.calculate_accuracy(train_predictions, ytrain)
            print "train accuracy: ", train_accuracy * 100
            for i in range(len(train_predictions)):
                train_predictions[i] = round(train_predictions[i])
            train_confMatrix = confusion_matrix(ytrain, train_predictions, labels = [1.0, 0.0])
            print "train confusion matrix:", train_confMatrix
            # predict test labels
            test_predictions = sm_results.predict(Xtest)
            test_accuracy = self.calculate_accuracy(test_predictions, ytest)
            print "test accuracy: ", test_accuracy * 100
            for i in range(len(test_predictions)):
                test_predictions[i] = round(test_predictions[i])
            test_confMatrix = confusion_matrix(ytest, test_predictions, labels = [1.0, 0.0])
            print "test confusion matrix:", test_confMatrix