diff --git a/examples/benchmarks/XGBoost/workflow_config_xgboost.yaml b/examples/benchmarks/XGBoost/workflow_config_xgboost.yaml index 31eee8206..398f8bd9e 100644 --- a/examples/benchmarks/XGBoost/workflow_config_xgboost.yaml +++ b/examples/benchmarks/XGBoost/workflow_config_xgboost.yaml @@ -30,14 +30,12 @@ task: module_path: qlib.contrib.model.xgboost kwargs: eval_metric: rmse - colsample_bytree: 0.5 - eta: 0.2 - gamma: 0.55 - max_depth: 2 - min_child_weight: 1.0 - n_estimators: 647 - subsample: 0.8 - nthread: 4 + colsample_bytree: 0.8879 + eta: 0.0421 + max_depth: 8 + n_estimators: 650 + subsample: 0.8789 + nthread: 20 dataset: class: DatasetH module_path: qlib.data.dataset @@ -62,4 +60,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