debugging confusion matrix

config/config.py

@@ -4,7 +4,7 @@ import argparse
 def load_args():
     parser = argparse.ArgumentParser()
 
-    parser.add_argument('--epoches', type=int, default=100)
+    parser.add_argument('--epoches', type=int, default=10)
     parser.add_argument('--save_inter', type=int, default=50)
     parser.add_argument('--eval_inter', type=int, default=1)
     parser.add_argument('--noise_threshold', type=int, default=0)

main.py

@@ -235,6 +235,7 @@ def make_prediction_duo(model, data, f_name):
     classes = data.dataset.dataset.classes
     print('Prediction made')
     # Build confusion matrix
+    print(len(y_true),len(y_pred))
     cf_matrix = confusion_matrix(y_true, y_pred)
     print('CM made')
     df_cm = pd.DataFrame(cf_matrix[:, None], index=[i for i in classes],