add pretrain-mode to gats

2026-07-01 18:11:18 +08:00 · 2020-11-25 17:46:34 +08:00
parent c14a99a735
commit cd7c81cfd0
5 changed files with 27 additions and 4 deletions
--- a/examples/benchmarks/GATs/worflow_config_gats.yaml
+++ b/examples/benchmarks/GATs/worflow_config_gats.yaml
@@ -37,9 +37,10 @@ task:
            lr: 1e-3
            early_stop: 20
            batch_size: 800
-            metric: IC
+            metric: loss
            loss: mse
-            base_model: GRU
+            base_model: LSTM
+            with_pretrain: True
            seed: 0
            GPU: 0
    dataset:
--- a/examples/benchmarks/GRU/model_gru_csi300.pkl
+++ b/examples/benchmarks/GRU/model_gru_csi300.pkl
--- a/examples/benchmarks/LSTM/model_lstm_csi300.pkl
+++ b/examples/benchmarks/LSTM/model_lstm_csi300.pkl
--- a/examples/workflow_by_code_gats.py
+++ b/examples/workflow_by_code_gats.py
@@ -70,9 +70,10 @@ if __name__ == "__main__":
                "lr": 1e-3,
                "early_stop": 20,
                "batch_size": 800,
-                "metric": "IC",
+                "metric": "loss",
                "loss": "mse",
-                "base_model": "GRU",
+                "base_model": "LSTM",
+                "with_pretrain": True,
                "seed": 0,
                "GPU": 0,
            },