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