Browse Source

fix ts api config (#2970)

* fix ts api config

* fix ts api config
Sunflower7788 9 months ago
parent
commit
851f809652

+ 3 - 3
paddlex/repo_apis/PaddleTS_api/configs/Nonstationary.yaml

@@ -24,10 +24,10 @@ info_params:
 model: 
   name: Nonstationary_Transformer
   model_cfg:
-    c_in: 1
-    factor: 3
+    c_in: 7
+    factor: 1
     p_hidden_dims: [256, 256]
     optimizer_params:
       learning_rate: 0.0001
       gamma: 0.5
-    patience: 5
+    patience: 3

+ 1 - 1
paddlex/repo_apis/PaddleTS_api/configs/Nonstationary_ad.yaml

@@ -25,7 +25,7 @@ info_params:
 model: 
   name: NonStationary_AD
   model_cfg:
-    c_in: 2
+    c_in: 5
     factor: 3
     p_hidden_dims: [32, 32]
     d_model: 64

+ 7 - 8
paddlex/repo_apis/PaddleTS_api/configs/PatchTST.yaml

@@ -25,17 +25,16 @@ info_params:
 model: 
   name: PatchTSTModel
   model_cfg:
-    c_in: 1
+    c_in: 7
     n_layers: 3
-    n_heads: 16
-    d_model: 128
-    d_ff: 256
-    dropout: 0.2
-    fc_dropout: 0.2
+    n_heads: 4
+    d_model: 16
+    d_ff: 128
+    dropout: 0.3
+    fc_dropout: 0.3
     head_dropout: 0.0
     patch_len: 16
     stride: 8
-    patience: 10
+    patience: 20
     optimizer_params:
       learning_rate: 0.0001
-      gamma: 0.9

+ 6 - 6
paddlex/repo_apis/PaddleTS_api/configs/TiDE.yaml

@@ -25,15 +25,15 @@ info_params:
 model: 
   name: TiDE
   model_cfg:
-    c_in: 1
+    c_in: 7
     num_encoder_layers: 2
-    use_revin: False
+    use_revin: True
     drop_prob: 0.5
-    hidden_size: 1024
-    decoder_output_dim: 8
-    temporal_decoder_hidden: 64
+    hidden_size: 512
+    decoder_output_dim: 32
+    temporal_decoder_hidden: 16
     optimizer_params:
-      learning_rate: 0.00099
+      learning_rate: 0.00098
       gamma: 0.5
     patience: 10
 

+ 2 - 2
paddlex/repo_apis/PaddleTS_api/configs/TimesNet.yaml

@@ -26,8 +26,8 @@ info_params:
 model: 
   name: TimesNetModel
   model_cfg:
-    c_in: 1 #
-    c_out: 1 #
+    c_in: 321 #
+    c_out: 321 #
     e_layers: 2 #
     num_kernels: 6 #
     d_model: 32 #

+ 7 - 5
paddlex/repo_apis/PaddleTS_api/configs/TimesNet_ad.yaml

@@ -25,10 +25,12 @@ info_params:
 model: 
   name: TimesNet_AD
   model_cfg:
-    c_in: 2
-    e_layers: 1
+    e_layers: 2
     num_kernels: 6
-    d_model: 8
+    d_model: 32
+    d_ff: 64
+    top_k: 3 
+    window_sampling_limit: Null
     optimizer_params:
-      learning_rate: 0.001
-      gamma: 0.5
+      learning_rate: 0.0005
+      gamma: 0.9

+ 3 - 3
paddlex/repo_apis/PaddleTS_api/configs/TimesNet_cls.yaml

@@ -28,15 +28,15 @@ info_params:
 model: 
   name: TimesNet_CLS
   model_cfg:
-    e_layers: 3 #
+    e_layers: 2 #
     num_kernels: 6 #
     d_model: 32 #
-    d_ff: 32 #
+    d_ff: 64 #
     top_k: 3 # 
     window_sampling_limit: Null # 
     patience: 10 #
     optimizer_params:
-      learning_rate: 0.001
+      learning_rate: 0.0005
       gamma: 0.9
 
 output: 'output/'