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])