From 6ddd3154f67b7699fe8922e868c2820179ba1095 Mon Sep 17 00:00:00 2001 From: Mahsa Date: Fri, 22 Dec 2023 14:05:35 +0200 Subject: [PATCH] Changing reinvent class input --- src/vai_lab/DataProcessing/plugins/reinvent.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/src/vai_lab/DataProcessing/plugins/reinvent.py b/src/vai_lab/DataProcessing/plugins/reinvent.py index 15274be6..3f7d74fa 100644 --- a/src/vai_lab/DataProcessing/plugins/reinvent.py +++ b/src/vai_lab/DataProcessing/plugins/reinvent.py @@ -71,14 +71,11 @@ def __init__(self, config = {}, data_in = [None], ini = False): class ReinventModel: - def __init__(self, seed=1, acquisition="uncertainty"): - self.seed = seed - self.acquisition = acquisition + def __init__(self): return - def do_run(self): - seed = self.seed - acquisition = self.acquisition + def do_run(self, seed=1, acquisition="uncertainty"): + ############################## # Quick options FIT_MODEL = True # whether to fit a Stan model or not