fix confusion matrix

main.py

@@ -236,9 +236,9 @@ def make_prediction_duo(model, data, f_name):
 
     # Build confusion matrix
     cf_matrix = confusion_matrix(y_true, y_pred)
-    df_cm = pd.DataFrame(cf_matrix / np.sum(cf_matrix, axis=1)[:, None], index=[i for i in classes],
+    df_cm = pd.DataFrame(cf_matrix[:, None], index=[i for i in classes],
                          columns=[i for i in classes])
-    plt.figure(figsize=(12, 7))
+    plt.figure(figsize=(14, 9))
     sn.heatmap(df_cm, annot=True)
     plt.savefig(f_name)