예제 #1
0
 def make_predictions(self, trained_model, data):
     X_train, X_test, y_train, y_test = data
     #get predictions
     print(type(trained_model))
     print("Making predictions!")
     train_pred = trained_model.predict(X_train)
     train_pred_prob = trained_model.predict_proba(X_train)
     test_pred = trained_model.predict(X_test)
     test_pred_prob = trained_model.predict_proba(X_test)
     print(test_pred_prob)
     # print metrics
     measurer = Measurement()
     print("Train resutls:")
     measurer.print_measurements(y_train, train_pred, train_pred_prob[:, 1])
     print("Test resutls:")
     measurer.print_measurements(y_test, test_pred, test_pred_prob[:, 1])