add pretrain-mode to gats

2026-07-02 18:40:58 +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