ensemble_model_prediction correction

kspruthviraj · Jul 13, 2022 · 5ef001c · 5ef001c
1 parent 8efc98d
commit 5ef001c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/model_training.py b/utils/model_training.py
@@ -413,7 +413,7 @@ def run_ensemble_prediction_on_unseen(self, test_main, data_loader, name):
         Ens_DEIT_label = Ens_DEIT_label.tolist()
 
         To_write = [i + '------------------' + j + '\n' for i, j in zip(im_names[0], Ens_DEIT_label)]
-        np.savetxt(test_main.params.test_outpath + '/Ensemble_models_Plankiformer_predictions' + name2 + name + '.txt', To_write, fmt='%s')
+        np.savetxt(test_main.params.test_outpath + '/Ensemble_models_Plankiformer_predictions_' + name2 + name + '.txt', To_write, fmt='%s')
 
     def initialize_model(self, train_main, test_main, data_loader, lr):
         device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")