diff --git a/main.py b/main.py index 29ae275d78657e223eee040122678c74e79c1748..ff27a40e75a7eb057e321c27cf81f770fff2ac10 100644 --- a/main.py +++ b/main.py @@ -236,6 +236,7 @@ def make_prediction_duo(model, data, f_name): print('Prediction made') # Build confusion matrix cf_matrix = confusion_matrix(y_true, y_pred) + print('CM made') df_cm = pd.DataFrame(cf_matrix[:, None], index=[i for i in classes], columns=[i for i in classes])