renamed to tuning

marnixkoops · Oct 28, 2021 · eb676c4 · eb676c4
1 parent eee54ef
commit eb676c4
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 4 deletions.
diff --git a/conf/base/parameters.yml b/conf/base/parameters.yml
@@ -40,7 +40,7 @@ train_tabnet:
         patience: 64
         batch_size: 256
         callbacks:
-            - class: kendrite.pipelines.tune_model.TuneReportCallback
+            - class: kendrite.pipelines.tuning.TuneReportCallback
               kwargs:
                   metrics: &monitor "valid_logloss"
                   "on": "epoch_end"

diff --git a/src/kendrite/pipeline_registry.py b/src/kendrite/pipeline_registry.py
@@ -3,7 +3,7 @@
 
 from kedro.pipeline import Pipeline, pipeline
 
-from kendrite.pipelines import engineering, evaluation, model, tune_model
+from kendrite.pipelines import engineering, evaluation, model, tuning
 
 
 def register_pipelines() -> Dict[str, Pipeline]:
@@ -16,7 +16,7 @@ def register_pipelines() -> Dict[str, Pipeline]:
     engineering_pipeline = engineering.create_data_pipeline()
     model_pipeline = model.create_model_pipeline()
     evaluation_pipeline = evaluation.create_evaluation_pipeline()
-    tune_pipeline = tune_model.create_pipeline()
+    tune_pipeline = tuning.create_pipeline()
     model_pipeline_from_tune = model.create_model_pipeline(
         from_hyperparameter_tuning=True
     )

diff --git a/src/kendrite/pipelines/tune_model/README.md → src/kendrite/pipelines/tuning/README.md b/src/kendrite/pipelines/tune_model/README.md → src/kendrite/pipelines/tuning/README.md
diff --git a/src/kendrite/pipelines/tune_model/nodes.py → src/kendrite/pipelines/tuning/nodes.py b/src/kendrite/pipelines/tune_model/nodes.py → src/kendrite/pipelines/tuning/nodes.py
diff --git a/...kendrite/pipelines/tune_model/pipeline.py → src/kendrite/pipelines/tuning/pipeline.py b/...kendrite/pipelines/tune_model/pipeline.py → src/kendrite/pipelines/tuning/pipeline.py
@@ -38,7 +38,7 @@ def create_pipeline(config_mapper: Callable = map_config_params) -> Pipeline:
                 patience: 64
                 batch_size: 256
                 callbacks:
-                    - class: kendrite.pipelines.tune_model.TuneReportCallback
+                    - class: kendrite.pipelines.tuning.TuneReportCallback
                       kwargs:
                           metrics: "val_logloss"
                           "on": 'epoch_end'