Add run_all_model script

2026-07-06 12:30:57 +08:00 · 2020-11-20 16:09:40 +08:00
parent 38e76a6e40
commit 5aa48524d6
13 changed files with 340 additions and 71 deletions
--- a/examples/benchmarks/CatBoost/workflow_config_catboost.yaml
+++ b/examples/benchmarks/CatBoost/workflow_config_catboost.yaml
@@ -1,4 +1,5 @@
 provider_uri: "~/.qlib/qlib_data/cn_data"
+region: cn
 market: &market csi300
 benchmark: &benchmark SH000300
 data_handler_config: &data_handler_config
--- a/examples/benchmarks/DNN/workflow_config_dnn.yaml
+++ b/examples/benchmarks/DNN/workflow_config_dnn.yaml
@@ -1,4 +1,5 @@
 provider_uri: "~/.qlib/qlib_data/cn_data"
+region: cn
 market: &market csi300
 benchmark: &benchmark SH000300
 data_handler_config: &data_handler_config
@@ -28,18 +29,16 @@ task:
        class: DNNModelPytorch
        module_path: qlib.contrib.model.pytorch_nn
        kwargs:
+            loss: mse
            input_dim: 360
            output_dim: 1
-            layers: [256, 512, 1024, 512, 256, 128, 64]
-            lr: 0.001
-            max_steps: 300
-            batch_size: 2000
-            early_stop_rounds: 50
-            eval_steps: 20
+            lr: 0.002
            lr_decay: 0.96
            lr_decay_steps: 100
-            optimizer: gd
-            loss: mse
+            optimizer: adam
+            max_steps: 8000
+            batch_size: 4096
+            GPU: 0
    dataset:
        class: DatasetH
        module_path: qlib.data.dataset
--- a/examples/benchmarks/GATs/worflow_config_gats.yaml
+++ b/examples/benchmarks/GATs/worflow_config_gats.yaml
@@ -1,4 +1,5 @@
 provider_uri: "~/.qlib/qlib_data/cn_data"
+region: cn
 market: &market csi300
 benchmark: &benchmark SH000300
 data_handler_config: &data_handler_config
--- a/examples/benchmarks/GBDT/workflow_config_gbdt.yaml
+++ b/examples/benchmarks/GBDT/workflow_config_gbdt.yaml
@@ -1,4 +1,5 @@
 provider_uri: "~/.qlib/qlib_data/cn_data"
+region: cn
 market: &market csi300
 benchmark: &benchmark SH000300
 data_handler_config: &data_handler_config
--- a/examples/benchmarks/GRU/workflow_config_gru.yaml
+++ b/examples/benchmarks/GRU/workflow_config_gru.yaml
@@ -1,4 +1,5 @@
 provider_uri: "~/.qlib/qlib_data/cn_data"
+region: cn
 market: &market csi300
 benchmark: &benchmark SH000300
 data_handler_config: &data_handler_config
--- a/examples/benchmarks/LSTM/workflow_config_lstm.yaml
+++ b/examples/benchmarks/LSTM/workflow_config_lstm.yaml
@@ -1,4 +1,5 @@
 provider_uri: "~/.qlib/qlib_data/cn_data"
+region: cn
 market: &market csi300
 benchmark: &benchmark SH000300
 data_handler_config: &data_handler_config
--- a/examples/benchmarks/XGBoost/workflow_config_xgboost.yaml
+++ b/examples/benchmarks/XGBoost/workflow_config_xgboost.yaml
@@ -1,4 +1,5 @@
 provider_uri: "~/.qlib/qlib_data/cn_data"
+region: cn
 market: &market csi300
 benchmark: &benchmark SH000300
 data_handler_config: &data_handler_config