From 5acd4c452ea04a2188c030e7fd4136a9bdbc5504 Mon Sep 17 00:00:00 2001 From: Insaf Ashrapov Date: Tue, 3 Oct 2023 23:01:27 +0300 Subject: [PATCH] decreasem estimators in boosting --- README.md | 2 +- pip_desc.md | 2 +- src/tabgan/adversarial_model.py | 2 +- src/tabgan/sampler.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index a7fb89e..6bdc70c 100644 --- a/README.md +++ b/README.md @@ -36,7 +36,7 @@ new_train4, new_target4 = GANGenerator(gen_x_times=1.1, cat_cols=None, bot_filter_quantile=0.001, top_filter_quantile=0.999, is_post_process=True, adversarial_model_params={ "metrics": "AUC", "max_depth": 2, "max_bin": 100, - "learning_rate": 0.02, "random_state": 42, "n_estimators": 500, + "learning_rate": 0.02, "random_state": 42, "n_estimators": 100, }, pregeneration_frac=2, only_generated_data=False, gen_params = {"batch_size": 500, "patience": 25, "epochs" : 500,}).generate_data_pipe(train, target, test, deep_copy=True, only_adversarial=False, use_adversarial=True) diff --git a/pip_desc.md b/pip_desc.md index a7fb89e..eecbd93 100644 --- a/pip_desc.md +++ b/pip_desc.md @@ -36,7 +36,7 @@ new_train4, new_target4 = GANGenerator(gen_x_times=1.1, cat_cols=None, bot_filter_quantile=0.001, top_filter_quantile=0.999, is_post_process=True, adversarial_model_params={ "metrics": "AUC", "max_depth": 2, "max_bin": 100, - "learning_rate": 0.02, "random_state": 42, "n_estimators": 500, + "learning_rate": 0.02, "random_state": 42, "n_estimators": 150, }, pregeneration_frac=2, only_generated_data=False, gen_params = {"batch_size": 500, "patience": 25, "epochs" : 500,}).generate_data_pipe(train, target, test, deep_copy=True, only_adversarial=False, use_adversarial=True) diff --git a/src/tabgan/adversarial_model.py b/src/tabgan/adversarial_model.py index 29efa04..934885f 100644 --- a/src/tabgan/adversarial_model.py +++ b/src/tabgan/adversarial_model.py @@ -98,7 +98,7 @@ def __init__( if model_params is None: self.model_params = { "metrics": "AUC", - "n_estimators": 5000, + "n_estimators": 150, "learning_rate": 0.04, "random_state": 42, } diff --git a/src/tabgan/sampler.py b/src/tabgan/sampler.py index 480c482..b912662 100644 --- a/src/tabgan/sampler.py +++ b/src/tabgan/sampler.py @@ -62,7 +62,7 @@ def __init__( "metrics": "AUC", "max_depth": 2, "max_bin": 100, - "n_estimators": 500, + "n_estimators": 150, "learning_rate": 0.02, "random_state": 42, },