From ebf482b84d3261e6e5314348250e667297bcfda2 Mon Sep 17 00:00:00 2001
From: liuht <liuht@lamda.nju.edu.cn>
Date: Sat, 13 Jan 2024 09:30:48 +0800
Subject: [PATCH] [FIX] delete unused lgb  params

---
 examples/dataset_table_workflow/config.py   | 17 -----------------
 examples/dataset_table_workflow/workflow.py |  4 ++--
 2 files changed, 2 insertions(+), 19 deletions(-)

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()