diff --git a/examples/dataset_table_workflow/config.py b/examples/dataset_table_workflow/config.py index a23cc31..7f0d5de 100644 --- a/examples/dataset_table_workflow/config.py +++ b/examples/dataset_table_workflow/config.py @@ -61,23 +61,6 @@ market_mapping_params = { } user_model_params = { - # "Corporacion": { - # "lgb": { - # "params": { - # "num_leaves": 31, - # "objective": "regression", - # "learning_rate": 0.1, - # "feature_fraction": 0.8, - # "bagging_fraction": 0.8, - # "bagging_freq": 2, - # "metric": "l2", - # "num_threads": 4, - # "verbose": -1, - # }, - # "MAX_ROUNDS": 500, - # "early_stopping_rounds": 50, - # } - # }, "M5": { "lgb": { "params": { diff --git a/examples/dataset_table_workflow/workflow.py b/examples/dataset_table_workflow/workflow.py index 73ebc86..5dbe80d 100644 --- a/examples/dataset_table_workflow/workflow.py +++ b/examples/dataset_table_workflow/workflow.py @@ -38,8 +38,8 @@ class TableDatasetWorkflow: workflow = HeterogeneousDatasetWorkflow( benchmark_config=hetero_cross_task_benchmark_config, name="hetero", - rebuild=True, - retrain=True + rebuild=False, + retrain=False ) workflow.labeled_hetero_table_example()