from sklearn.metrics import classification_report target_names = ['first_value_y','second_value_y'] # target values # Print classification report after a train/test split: print(classification_report(y_test, y_pred, target_names=target_names))