From f3446ce0b0806128e3ac7da20d47dd96c046b6fc Mon Sep 17 00:00:00 2001 From: Pavel Zwerschke Date: Mon, 20 Mar 2023 10:46:02 +0100 Subject: [PATCH 1/2] Make lgbm large actually large --- benchmark.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/benchmark.py b/benchmark.py index 88ab3f9..17ff367 100644 --- a/benchmark.py +++ b/benchmark.py @@ -53,7 +53,8 @@ def train_gbdt_lgbm() -> lgb.LGBMRegressor: def train_gbdt_large_lgbm() -> lgb.LGBMRegressor: return load_model( - "gbdt_large_lgbm", lambda: lgb.LGBMRegressor(n_estimators=2000, random_state=42) + "gbdt_large_lgbm", + lambda: lgb.LGBMRegressor(n_estimators=20000, random_state=42), ) From c73bf1ff2f40ba76ce5ac5440f236f9b5a5de947 Mon Sep 17 00:00:00 2001 From: Pavel Zwerschke Date: Mon, 20 Mar 2023 12:07:10 +0100 Subject: [PATCH 2/2] load_rf_lgbm -> train_rf_lgbm --- benchmark.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/benchmark.py b/benchmark.py index 17ff367..012dd8d 100644 --- a/benchmark.py +++ b/benchmark.py @@ -58,7 +58,7 @@ def train_gbdt_large_lgbm() -> lgb.LGBMRegressor: ) -def load_rf_lgbm() -> lgb.LGBMRegressor: +def train_rf_lgbm() -> lgb.LGBMRegressor: return load_model( "rg_lgbm", lambda: lgb.LGBMRegressor( @@ -68,6 +68,7 @@ def load_rf_lgbm() -> lgb.LGBMRegressor: random_state=42, bagging_freq=5, bagging_fraction=0.5, + verbose=-1, ), ) @@ -218,8 +219,8 @@ def loads_lzma(data): ("LGBM gbdt LZMA", train_gbdt_lgbm) + dumps_lgbm_lzma_args, ("LGBM gbdt large", train_gbdt_large_lgbm) + dumps_lgbm_args, ("LGBM gbdt large LZMA", train_gbdt_large_lgbm) + dumps_lgbm_lzma_args, - ("LGBM rf", load_rf_lgbm) + dumps_lgbm_args, - ("LGBM rf LZMA", load_rf_lgbm) + dumps_lgbm_lzma_args, + ("LGBM rf", train_rf_lgbm) + dumps_lgbm_args, + ("LGBM rf LZMA", train_rf_lgbm) + dumps_lgbm_lzma_args, ] benchmark_results = [benchmark_model(*args) for args in models_to_benchmark] print("Base results / Our results / Change")