diff --git a/examples/benchmarks/CatBoost/workflow_config_catboost.yaml b/examples/benchmarks/CatBoost/workflow_config_catboost.yaml index 8bf3bb72b..574b52ddd 100644 --- a/examples/benchmarks/CatBoost/workflow_config_catboost.yaml +++ b/examples/benchmarks/CatBoost/workflow_config_catboost.yaml @@ -30,8 +30,9 @@ task: module_path: qlib.contrib.model.catboost_model kwargs: loss: RMSE - iterations: 5 - learning_rate: 0.03 + learning_rate: 0.0421 + subsample: 0.8789 + thread_count: 20 dataset: class: DatasetH module_path: qlib.data.dataset @@ -56,4 +57,4 @@ task: - class: PortAnaRecord module_path: qlib.workflow.record_temp kwargs: - config: *port_analysis_config \ No newline at end of file + config: *port_analysis_config