Browse Source

[FIX] delete unused lgb params

tags/v0.3.2
liuht 2 years ago
parent
commit
ebf482b84d
2 changed files with 2 additions and 19 deletions
  1. +0
    -17
      examples/dataset_table_workflow/config.py
  2. +2
    -2
      examples/dataset_table_workflow/workflow.py

+ 0
- 17
examples/dataset_table_workflow/config.py View File

@@ -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": {


+ 2
- 2
examples/dataset_table_workflow/workflow.py View File

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



Loading…
Cancel
Save