update the default for compression and add some better logging

ryan-wolbeck · Aug 14, 2020 · 68f9a41 · 68f9a41
1 parent 8c9ccc3
commit 68f9a41
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/ngboost_tuner/cli.py b/ngboost_tuner/cli.py
@@ -54,7 +54,7 @@ def build_cli():
         "-ct",
         "--compression-type",
         type=str,
-        default=os.getenv("COMPRESSION"),
+        default=os.getenv("COMPRESSION", None),
         help="Input data compression, i.e. gzip or None; defaults to $COMPRESSION if not set",
     )
     tune.add_argument(

diff --git a/ngboost_tuner/tune.py b/ngboost_tuner/tune.py
@@ -233,6 +233,8 @@ def objective(params):
             Y_val=y_valid.values,
             early_stopping_rounds=2,
         )
+        log.info(f"Final LightGBM parameters {final_params_lightgbm}")
+        log.info(f"Final NGBoost parameters {final_ngboost_params}")
 
     else:
         base_models = [