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")