From a9c1f8c2a0dd86adfc1683d82bfe6acb6d2e5497 Mon Sep 17 00:00:00 2001 From: Haoyu Wang <48108414+javaThonc@users.noreply.github.com> Date: Fri, 27 Nov 2020 17:16:35 +0800 Subject: [PATCH] Update workflow_config_xgboost.yaml --- .../XGBoost/workflow_config_xgboost.yaml | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) 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