From a67a6134b4d281eaa37506573588b507e528754a Mon Sep 17 00:00:00 2001 From: Young Date: Wed, 10 Jul 2024 05:34:09 +0000 Subject: [PATCH] We must align with previous results --- examples/benchmarks/GeneralPtNN/workflow_config_gru.yaml | 4 ++-- examples/benchmarks/GeneralPtNN/workflow_config_mlp.yaml | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/examples/benchmarks/GeneralPtNN/workflow_config_gru.yaml b/examples/benchmarks/GeneralPtNN/workflow_config_gru.yaml index a2f03a230..9c4024e31 100755 --- a/examples/benchmarks/GeneralPtNN/workflow_config_gru.yaml +++ b/examples/benchmarks/GeneralPtNN/workflow_config_gru.yaml @@ -52,8 +52,8 @@ port_analysis_config: &port_analysis_config min_cost: 5 task: model: - class: GRU - module_path: qlib.contrib.model.pytorch_gru_ts + class: GeneralPTNN + module_path: qlib.contrib.model.pytorch_general_nn kwargs: d_feat: 20 hidden_size: 64 diff --git a/examples/benchmarks/GeneralPtNN/workflow_config_mlp.yaml b/examples/benchmarks/GeneralPtNN/workflow_config_mlp.yaml index 6c85546ca..0ac3297fd 100644 --- a/examples/benchmarks/GeneralPtNN/workflow_config_mlp.yaml +++ b/examples/benchmarks/GeneralPtNN/workflow_config_mlp.yaml @@ -57,8 +57,8 @@ port_analysis_config: &port_analysis_config min_cost: 5 task: model: - class: DNNModelPytorch - module_path: qlib.contrib.model.pytorch_nn + class: GeneralPTNN + module_path: qlib.contrib.model.pytorch_general_nn kwargs: loss: mse lr: 0.002