From 5ef001c9873c415b72ef2211da63849241ebbd8a Mon Sep 17 00:00:00 2001 From: Sreenath P Kyathanahally Date: Wed, 13 Jul 2022 22:21:50 +0200 Subject: [PATCH] ensemble_model_prediction correction --- utils/model_training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/model_training.py b/utils/model_training.py index 1334e09..8b46322 100644 --- 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")