diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter amalonaticus/CITAMA5_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter amalonaticus/CITAMA5_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter amalonaticus/CITAMA5_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter amalonaticus/CITAMA5_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter amalonaticus/CITAMA5_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter amalonaticus/CITAMA5_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter amalonaticus/CITAMA5_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter amalonaticus/CITAMA5_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter amalonaticus/CITAMA6_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter amalonaticus/CITAMA6_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter amalonaticus/CITAMA6_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter amalonaticus/CITAMA6_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter amalonaticus/CITAMA6_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter amalonaticus/CITAMA6_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter amalonaticus/CITAMA6_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter amalonaticus/CITAMA6_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter braakii/CITBRA5_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter braakii/CITBRA5_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter braakii/CITBRA5_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter braakii/CITBRA5_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter braakii/CITBRA5_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter braakii/CITBRA5_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter braakii/CITBRA5_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter braakii/CITBRA5_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter cronae/CITCRO1_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter cronae/CITCRO1_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter cronae/CITCRO1_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter cronae/CITCRO1_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter cronae/CITCRO1_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter cronae/CITCRO1_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter cronae/CITCRO1_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter cronae/CITCRO1_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter cronae/CITCRO2_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter cronae/CITCRO2_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter cronae/CITCRO2_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter cronae/CITCRO2_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter cronae/CITCRO2_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter cronae/CITCRO2_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter cronae/CITCRO2_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter cronae/CITCRO2_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter cronae/CITCRO3_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter cronae/CITCRO3_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter cronae/CITCRO3_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter cronae/CITCRO3_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter cronae/CITCRO3_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter cronae/CITCRO3_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter cronae/CITCRO3_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter cronae/CITCRO3_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter farmeri/CITFAR2_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter farmeri/CITFAR2_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter farmeri/CITFAR2_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter farmeri/CITFAR2_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter farmeri/CITFAR2_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter farmeri/CITFAR2_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter farmeri/CITFAR2_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter farmeri/CITFAR2_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter farmeri/CITFAR4_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter farmeri/CITFAR4_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter farmeri/CITFAR4_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter farmeri/CITFAR4_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter farmeri/CITFAR4_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter farmeri/CITFAR4_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter farmeri/CITFAR4_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter farmeri/CITFAR4_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE17_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE17_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE17_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE17_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE19_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE19_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE19_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE19_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE19_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE19_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE19_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE19_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE21_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE21_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE21_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE21_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE21_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE21_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE21_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE21_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE22_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE22_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE22_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE22_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE22_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE22_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE22_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE22_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE24_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE24_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE24_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE24_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE24_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE24_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE24_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE24_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE26_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE26_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE26_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE26_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE26_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE26_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE26_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE26_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE27_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE27_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE27_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE27_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE27_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE27_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE27_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter freundii/CITFRE27_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter koseri/CITKOS13_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter koseri/CITKOS13_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter koseri/CITKOS13_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter koseri/CITKOS13_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter koseri/CITKOS13_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter koseri/CITKOS13_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter koseri/CITKOS13_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter koseri/CITKOS13_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter koseri/CITKOS18_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter koseri/CITKOS18_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter koseri/CITKOS18_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter koseri/CITKOS18_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter koseri/CITKOS18_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter koseri/CITKOS18_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter koseri/CITKOS18_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter koseri/CITKOS18_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter koseri/CITKOS21_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter koseri/CITKOS21_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter koseri/CITKOS21_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter koseri/CITKOS21_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter koseri/CITKOS21_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter koseri/CITKOS21_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter koseri/CITKOS21_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter koseri/CITKOS21_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter portucalensis/CITPOR3_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter portucalensis/CITPOR3_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter portucalensis/CITPOR3_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter portucalensis/CITPOR3_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter portucalensis/CITPOR3_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter portucalensis/CITPOR3_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter portucalensis/CITPOR3_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter portucalensis/CITPOR3_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter portucalensis/CITPOR4_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter portucalensis/CITPOR4_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter portucalensis/CITPOR4_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter portucalensis/CITPOR4_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter portucalensis/CITPOR4_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Citrobacter portucalensis/CITPOR4_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Citrobacter portucalensis/CITPOR4_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Citrobacter portucalensis/CITPOR4_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter chengduensis/ENTCHE1_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter chengduensis/ENTCHE1_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter chengduensis/ENTCHE1_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter chengduensis/ENTCHE1_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter chengduensis/ENTCHE1_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter chengduensis/ENTCHE1_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter chengduensis/ENTCHE1_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter chengduensis/ENTCHE1_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter chengduensis/ENTCHE2_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter chengduensis/ENTCHE2_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter chengduensis/ENTCHE2_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter chengduensis/ENTCHE2_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter chengduensis/ENTCHE2_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter chengduensis/ENTCHE2_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter chengduensis/ENTCHE2_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter chengduensis/ENTCHE2_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter cloacae/ENTCLO88_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter cloacae/ENTCLO88_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter cloacae/ENTCLO88_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter cloacae/ENTCLO88_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter cloacae/ENTCLO88_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter cloacae/ENTCLO88_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter cloacae/ENTCLO88_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter cloacae/ENTCLO88_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter cloacae/ENTCLOCO22_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter cloacae/ENTCLOCO22_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter cloacae/ENTCLOCO22_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter cloacae/ENTCLOCO22_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter cloacae/ENTCLOCO22_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter cloacae/ENTCLOCO22_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter cloacae/ENTCLOCO22_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter cloacae/ENTCLOCO22_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR10_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR10_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR10_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR10_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR10_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR10_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR10_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR10_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR11_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR11_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR11_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR11_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR11_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR11_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR11_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR11_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR13_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR13_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR13_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR13_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR13_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR13_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR13_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR13_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR15_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR15_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR15_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR15_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR15_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR15_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR15_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR15_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR18_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR18_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR18_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR18_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR18_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR18_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR18_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR18_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR21_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR21_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR21_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR21_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR21_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR21_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR21_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR21_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR22_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR22_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR22_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR22_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR22_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR22_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR22_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR22_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR25_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR25_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR25_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR25_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR25_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR25_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR25_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR25_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR9_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR9_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR9_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR9_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR9_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR9_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter hormaechei/ENTHOR9_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter hormaechei/ENTHOR9_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter kobei/ENTKOB2_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter kobei/ENTKOB2_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter kobei/ENTKOB2_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter kobei/ENTKOB2_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Enterobacter kobei/ENTKOB2_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Enterobacter kobei/ENTKOB2_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Enterobacter kobei/ENTKOB2_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Enterobacter kobei/ENTKOB2_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Escherichia coli/ESCCOL283_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Escherichia coli/ESCCOL283_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Escherichia coli/ESCCOL283_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Escherichia coli/ESCCOL283_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Escherichia coli/ESCCOL283_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Escherichia coli/ESCCOL283_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Escherichia coli/ESCCOL283_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Escherichia coli/ESCCOL283_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Escherichia coli/ESCCOL284_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Escherichia coli/ESCCOL284_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Escherichia coli/ESCCOL284_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Escherichia coli/ESCCOL284_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Escherichia coli/ESCCOL284_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Escherichia coli/ESCCOL284_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Escherichia coli/ESCCOL284_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Escherichia coli/ESCCOL284_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Hafnia alvei/HAFALV13_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Hafnia alvei/HAFALV13_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Hafnia alvei/HAFALV13_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Hafnia alvei/HAFALV13_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Hafnia alvei/HAFALV13_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Hafnia alvei/HAFALV13_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Hafnia alvei/HAFALV13_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Hafnia alvei/HAFALV13_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE23_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE23_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE23_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE23_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE23_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE23_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE23_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE23_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE33_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE33_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE33_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE33_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE33_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE33_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE33_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE33_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE34_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE34_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE34_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE34_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE34_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE34_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE34_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE34_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE38_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE38_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE38_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE38_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE38_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE38_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE38_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE38_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE39_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE39_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE39_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE39_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE39_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE39_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE39_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE39_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE40_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE40_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE40_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE40_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE40_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE40_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE40_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE40_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE42_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE42_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE42_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE42_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE42_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE42_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE42_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE42_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE44_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE44_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE44_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE44_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE44_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE44_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE44_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE44_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE45_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE45_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE45_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE45_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE45_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE45_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE45_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE45_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE46_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE46_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE46_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE46_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE46_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE46_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE46_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE46_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE47_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE47_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE47_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE47_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE47_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE47_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE47_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE47_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE48_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE48_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE48_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE48_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE48_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE48_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE48_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE48_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE49_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE49_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE49_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE49_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE49_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE49_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE49_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE49_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE50_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE50_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE50_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE50_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE50_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE50_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE50_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE50_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE53_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE53_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE53_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE53_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE53_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE53_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella pneumoniae/KLEPNE53_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella pneumoniae/KLEPNE53_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella variicola/KLEVAR5_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella variicola/KLEVAR5_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella variicola/KLEVAR5_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella variicola/KLEVAR5_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella variicola/KLEVAR5_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella variicola/KLEVAR5_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella variicola/KLEVAR5_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella variicola/KLEVAR5_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella variicola/KLEVAR9_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella variicola/KLEVAR9_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella variicola/KLEVAR9_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella variicola/KLEVAR9_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Klebsiella variicola/KLEVAR9_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Klebsiella variicola/KLEVAR9_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Klebsiella variicola/KLEVAR9_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Klebsiella variicola/KLEVAR9_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG11_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG11_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG11_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG11_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG11_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG11_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG11_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG11_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG13_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG13_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG13_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG13_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG13_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG13_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG13_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG13_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG19_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG19_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG19_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG19_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG19_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG19_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG19_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG19_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG26_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG26_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG26_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG26_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG26_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG26_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Morganella morganii/MORMOG26_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Morganella morganii/MORMOG26_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR14_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR14_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR14_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR14_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR14_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR14_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR14_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR14_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR16_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR16_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR16_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR16_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR16_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR16_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR16_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR16_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR17_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR17_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR17_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR17_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR17_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR17_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR17_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR17_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR19_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR19_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR19_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR19_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR19_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR19_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR19_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR19_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR23_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR23_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR23_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR23_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR23_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR23_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR23_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Proteus mirabilis/PRTMIR23_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Providencia stuartii/PRVSTU1_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Providencia stuartii/PRVSTU1_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Providencia stuartii/PRVSTU1_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Providencia stuartii/PRVSTU1_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Providencia stuartii/PRVSTU1_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Providencia stuartii/PRVSTU1_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Providencia stuartii/PRVSTU1_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Providencia stuartii/PRVSTU1_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Serratia marcescens/SERMAR14_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Serratia marcescens/SERMAR14_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Serratia marcescens/SERMAR14_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Serratia marcescens/SERMAR14_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Serratia marcescens/SERMAR14_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Serratia marcescens/SERMAR14_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Serratia marcescens/SERMAR14_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Serratia marcescens/SERMAR14_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Serratia marcescens/SERMAR15_AER.npy b/data/processed_data/npy_image/data_test_contrastive/Serratia marcescens/SERMAR15_AER.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Serratia marcescens/SERMAR15_AER.npy
rename to data/processed_data/npy_image/data_test_contrastive/Serratia marcescens/SERMAR15_AER.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Serratia marcescens/SERMAR15_ANA.npy b/data/processed_data/npy_image/data_test_contrastive/Serratia marcescens/SERMAR15_ANA.npy
similarity index 100%
rename from data/processed_data/npy_image/data_training_contrastive/Serratia marcescens/SERMAR15_ANA.npy
rename to data/processed_data/npy_image/data_test_contrastive/Serratia marcescens/SERMAR15_ANA.npy
diff --git a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE17_AER.npy b/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE17_AER.npy
deleted file mode 100644
index 7624c5c77f9731533e1e58df8a5a982e0dcbcdc3..0000000000000000000000000000000000000000
Binary files a/data/processed_data/npy_image/data_training_contrastive/Citrobacter freundii/CITFRE17_AER.npy and /dev/null differ
diff --git a/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR27_AER.npy b/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR27_AER.npy
deleted file mode 100644
index 0848953df5dd01ab5d651ba21a089711cbe2453c..0000000000000000000000000000000000000000
Binary files a/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR27_AER.npy and /dev/null differ
diff --git a/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR27_ANA.npy b/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR27_ANA.npy
deleted file mode 100644
index f2a56b8a141ab2b9f9fbe498214003929d172a80..0000000000000000000000000000000000000000
Binary files a/data/processed_data/npy_image/data_training_contrastive/Proteus mirabilis/PRTMIR27_ANA.npy and /dev/null differ
diff --git a/image_ref/config.py b/image_ref/config.py
index a63c21726c266d0ed7f134a8952f3d62bf4a5213..3676bcdc2f4702c4d0578e750798b8f69a7a4662 100644
--- a/image_ref/config.py
+++ b/image_ref/config.py
@@ -12,7 +12,8 @@ def load_args_contrastive():
     parser.add_argument('--batch_size', type=int, default=64)
     parser.add_argument('--positive_prop', type=int, default=None)
     parser.add_argument('--model', type=str, default='ResNet18')
-    parser.add_argument('--dataset_dir', type=str, default='data/processed_data/npy_image/data_training_contrastive')
+    parser.add_argument('--dataset_train_dir', type=str, default='data/processed_data/npy_image/data_training_contrastive')
+    parser.add_argument('--dataset_val_dir', type=str, default='data/processed_data/npy_image/data_test_contrastive')
     parser.add_argument('--dataset_ref_dir', type=str, default='image_ref/img_ref')
     parser.add_argument('--output', type=str, default='output/out_contrastive.csv')
     parser.add_argument('--save_path', type=str, default='output/best_model_constrastive.pt')
diff --git a/image_ref/dataset_ref.py b/image_ref/dataset_ref.py
index f0f329ce1b9a15fc3471e426d03daa94164f9dc6..0fcfbdf70e9768b1f4a0a31ef588ef5396ae17e3 100644
--- a/image_ref/dataset_ref.py
+++ b/image_ref/dataset_ref.py
@@ -149,7 +149,8 @@ class ImageFolderDuo(data.Dataset):
     def __len__(self):
         return len(self.imlist)
 
-def load_data_duo(base_dir, batch_size, shuffle=True, noise_threshold=0, ref_dir = None, positive_prop=None):
+def load_data_duo(base_dir_train, base_dir_test, batch_size, shuffle=True, noise_threshold=0, ref_dir = None, positive_prop=None):
+
     train_transform = transforms.Compose(
         [transforms.Resize((224, 224)),
          Threshold_noise(noise_threshold),
@@ -164,13 +165,8 @@ def load_data_duo(base_dir, batch_size, shuffle=True, noise_threshold=0, ref_dir
          transforms.Normalize(0.5, 0.5)])
     print('Default val transform')
 
-    train_dataset = ImageFolderDuo(root=base_dir, transform=train_transform, ref_dir = ref_dir, positive_prop=positive_prop)
-    val_dataset = ImageFolderDuo(root=base_dir, transform=val_transform, ref_dir = ref_dir, positive_prop=positive_prop)
-    generator1 = torch.Generator().manual_seed(42)
-    indices = torch.randperm(len(train_dataset), generator=generator1)
-    val_size = len(train_dataset) // 5
-    train_dataset = torch.utils.data.Subset(train_dataset, indices[:-val_size])
-    val_dataset = torch.utils.data.Subset(val_dataset, indices[-val_size:])
+    train_dataset = ImageFolderDuo(root=base_dir_train, transform=train_transform, ref_dir = ref_dir, positive_prop=positive_prop)
+    val_dataset = ImageFolderDuo_Batched(root=base_dir_test, transform=val_transform, ref_dir = ref_dir)
 
     data_loader_train = data.DataLoader(
         dataset=train_dataset,
@@ -183,7 +179,7 @@ def load_data_duo(base_dir, batch_size, shuffle=True, noise_threshold=0, ref_dir
 
     data_loader_test = data.DataLoader(
         dataset=val_dataset,
-        batch_size=batch_size,
+        batch_size=1,
         shuffle=shuffle,
         num_workers=0,
         collate_fn=None,
@@ -231,47 +227,4 @@ class ImageFolderDuo_Batched(data.Dataset):
         return batched_imgAER, batched_imgANA, batched_im_ref, batched_label
 
     def __len__(self):
-        return len(self.imlist)
-
-def load_data_duo_batched(base_dir, shuffle=True, noise_threshold=0, ref_dir = None):
-    train_transform = transforms.Compose(
-        [transforms.Resize((224, 224)),
-         Threshold_noise(noise_threshold),
-         Log_normalisation(),
-         transforms.Normalize(0.5, 0.5)])
-    print('Default train transform')
-
-    val_transform = transforms.Compose(
-        [transforms.Resize((224, 224)),
-         Threshold_noise(noise_threshold),
-         Log_normalisation(),
-         transforms.Normalize(0.5, 0.5)])
-    print('Default val transform')
-
-    train_dataset = ImageFolderDuo_Batched(root=base_dir, transform=train_transform, ref_dir = ref_dir)
-    val_dataset = ImageFolderDuo_Batched(root=base_dir, transform=val_transform, ref_dir = ref_dir)
-    generator1 = torch.Generator().manual_seed(42)
-    indices = torch.randperm(len(train_dataset), generator=generator1)
-    val_size = len(train_dataset) // 5
-    train_dataset = torch.utils.data.Subset(train_dataset, indices[:-val_size])
-    val_dataset = torch.utils.data.Subset(val_dataset, indices[-val_size:])
-
-    data_loader_train = data.DataLoader(
-        dataset=train_dataset,
-        batch_size=1,
-        shuffle=shuffle,
-        num_workers=0,
-        collate_fn=None,
-        pin_memory=False,
-    )
-
-    data_loader_test = data.DataLoader(
-        dataset=val_dataset,
-        batch_size=1,
-        shuffle=shuffle,
-        num_workers=0,
-        collate_fn=None,
-        pin_memory=False,
-    )
-
-    return data_loader_train, data_loader_test
+        return len(self.imlist)
\ No newline at end of file
diff --git a/image_ref/main.py b/image_ref/main.py
index 82cd816a4192f96e5f3daf5314ee1df043bb2042..dba220ad7967946ee559295f515fd575d3eb79e1 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -2,7 +2,7 @@ import matplotlib.pyplot as plt
 import numpy as np
 
 from config import load_args_contrastive
-from dataset_ref import load_data_duo_batched, load_data_duo
+from dataset_ref import load_data_duo
 import torch
 import torch.nn as nn
 from model import Classification_model_contrastive, Classification_model_duo_contrastive
@@ -58,7 +58,7 @@ def test_duo(model, data_test, loss_function, epoch):
             img_ref = img_ref.cuda()
             label = label.cuda()
         pred_logits = model.forward(imaer,imana,img_ref)
-        pred_class = torch.argmax(pred_logits,dim=1)
+        pred_class = torch.argmax(pred_logits[:,0],dim=1)
         acc += (pred_class==label).sum().item()
         loss = loss_function(pred_logits,label)
         losses += loss.item()
@@ -69,10 +69,9 @@ def test_duo(model, data_test, loss_function, epoch):
 
 def run_duo(args):
     #load data
-    data_train, data_test = load_data_duo(base_dir=args.dataset_dir, batch_size=args.batch_size,
+    data_train, data_test_batch = load_data_duo(base_dir_train=args.dataset_train_dir, base_dir_test=args.dataset_val_dir, batch_size=args.batch_size,
                                           ref_dir=args.dataset_ref_dir, positive_prop=args.positive_prop)
-    data_train_batch, data_test_batch = load_data_duo_batched(base_dir=args.dataset_dir,
-                                                              ref_dir=args.dataset_ref_dir)
+
     #load model
     model = Classification_model_duo_contrastive(model = args.model, n_class=2)
     model.double()
@@ -157,10 +156,11 @@ def make_prediction_duo(model, data, f_name, f_name2):
             label = label.cuda()
         output = model(imaer,imana,img_ref)
         confidence_pred_list[specie].append(output[:,0].data.cpu().numpy())
-        output = (torch.max(torch.exp(output), 1)[1]).data.cpu().numpy()
+        #Mono class output (only most postive paire
+        output = torch.argmax(output[:,0],dim=1).data.cpu().numpy()
         y_pred.extend(output)
 
-        label = label.data.cpu().numpy()
+        label = torch.argmin(label).data.cpu().numpy()
         y_true.extend(label)  # Save Truth
     # constant for classes
 
@@ -199,4 +199,4 @@ def load_model(model, path):
 
 if __name__ == '__main__':
     args = load_args_contrastive()
-    run_duo(args)
+    run_duo(args)
\ No newline at end of file