updated mapped params and made ht shorter

marnixkoops · Oct 28, 2021 · 2626df8 · 2626df8
1 parent f977032
commit 2626df8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/conf/base/parameters.yml b/conf/base/parameters.yml
@@ -80,7 +80,7 @@ train_tabnet:
             class: ray.tune.suggest.optuna.OptunaSearch
             kwargs: {}
         # callbacks: - List[Dict]. Each dict-> class and kwarg: Dict[str,str]
-        num_samples: 200
+        num_samples: 20
         verbose: 1 # level of messages to print
         stop:
             training_iteration: *epoch

diff --git a/src/kendrite/pipelines/tune_model/nodes.py b/src/kendrite/pipelines/tune_model/nodes.py
@@ -90,7 +90,7 @@ def create_tabnet_trainer(
 
     def tabnet_trainer(config):
         mapped_params = config_mapper(params, config)
-        model = neural_model(params["estimator"])
+        model = neural_model(mapped_params["estimator"])
         fit(
             model, X_train, y_train, X_valid, y_valid, mapped_params["fit"], tuning=True
         )