fix(Bagging): Fixed incorrect DataGenerator parameter mapping in pred…

…ict function
frankkramer-lab · May 21, 2022 · 275b0b8 · 275b0b8
1 parent 00cbdf1
commit 275b0b8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/aucmedi/ensemble/bagging.py b/aucmedi/ensemble/bagging.py
@@ -401,10 +401,10 @@ def __prediction_process__(queue, model, path_model, datagen_paras):
                                 labels=None,
                                 metadata=datagen_paras["metadata"],
                                 batch_size=datagen_paras["batch_size"],
-                                data_aug=datagen_paras["metadata"],
+                                data_aug=datagen_paras["data_aug"],
                                 seed=datagen_paras["seed"],
                                 subfunctions=datagen_paras["subfunctions"],
-                                shuffle=False,
+                                shuffle=datagen_paras["shuffle"],
                                 standardize_mode=datagen_paras["standardize_mode"],
                                 resize=datagen_paras["resize"],
                                 grayscale=datagen_paras["grayscale"],