decreasem estimators in boosting

Diyago · Oct 3, 2023 · 5acd4c4 · 5acd4c4
1 parent c82255d
commit 5acd4c4
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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,
             },