docs: update tuning configs;

WenjieDu · May 28, 2024 · e58543b · e58543b
1 parent 5126271
commit e58543b
Show file tree

Hide file tree

Showing 28 changed files with 48 additions and 46 deletions.
diff --git a/benchmark_code/PyPOTS_tuning_configs/Autoformer/Autoformer_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/Autoformer/Autoformer_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.Autoformer --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.Autoformer --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/...g_configs/CSDI/CSDI_AQI_tuning_space.json → ...gs/CSDI/CSDI_BeijingAir_tuning_space.json b/...g_configs/CSDI/CSDI_AQI_tuning_space.json → ...gs/CSDI/CSDI_BeijingAir_tuning_space.json
diff --git a/benchmark_code/PyPOTS_tuning_configs/CSDI/CSDI_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/CSDI/CSDI_searching_config.yml
@@ -3,7 +3,7 @@ authorName: WenjieDu
 trialConcurrency: 1
 trainingServicePlatform: local
 #searchSpacePath: CSDI_PhysioNet2012_tuning_space.json
-#searchSpacePath: CSDI_AQI_tuning_space.json
+#searchSpacePath: CSDI_BeijingAir_tuning_space.json
 searchSpacePath: CSDI_ETTh1_tuning_space.json
 multiThread: true
 useAnnotation: false

diff --git a/benchmark_code/PyPOTS_tuning_configs/Crossformer/Crossformer_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/Crossformer/Crossformer_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.Crossformer --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.Crossformer --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/benchmark_code/PyPOTS_tuning_configs/DLinear/DLinear_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/DLinear/DLinear_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.DLinear --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.DLinear --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/benchmark_code/PyPOTS_tuning_configs/ETSformer/ETSformer_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/ETSformer/ETSformer_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.DLinear --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.DLinear --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/benchmark_code/PyPOTS_tuning_configs/FiLM/FiLM_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/FiLM/FiLM_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.FiLM --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.FiLM --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/benchmark_code/PyPOTS_tuning_configs/FreTS/FreTS_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/FreTS/FreTS_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.FreTS --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.FreTS --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/benchmark_code/PyPOTS_tuning_configs/GRUD/GRUD_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/GRUD/GRUD_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.GRUD --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.GRUD --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/benchmark_code/PyPOTS_tuning_configs/Informer/Informer_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/Informer/Informer_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.Informer --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.Informer --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/benchmark_code/PyPOTS_tuning_configs/Koopa/Koopa_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/Koopa/Koopa_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.Koopa --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.Koopa --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/benchmark_code/PyPOTS_tuning_configs/MICN/MICN_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/MICN/MICN_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.MICN --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.MICN --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/...g_configs/MRNN/MRNN_AQI_tuning_space.json → ...gs/MRNN/MRNN_BeijingAir_tuning_space.json b/...g_configs/MRNN/MRNN_AQI_tuning_space.json → ...gs/MRNN/MRNN_BeijingAir_tuning_space.json
diff --git a/benchmark_code/PyPOTS_tuning_configs/MRNN/MRNN_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/MRNN/MRNN_searching_config.yml
@@ -3,17 +3,17 @@ authorName: WenjieDu
 trialConcurrency: 1
 trainingServicePlatform: local
 searchSpacePath: MRNN_PhysioNet2012_tuning_space.json
-#searchSpacePath: MRNN_AQI_tuning_space.json
+#searchSpacePath: MRNN_BeijingAir_tuning_space.json
 #searchSpacePath: MRNN_ETTh1_tuning_space.json
 multiThread: true
 useAnnotation: false
 tuner:
     builtinTunerName: Random
 
 trial:
-    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.MRNN --train_set ../../data/physionet_2012/train.h5 --val_set ../../data/physionet_2012/val.h5 --test_set ../../data/physionet_2012/test.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.MRNN --train_set ../../data/physionet_2012/train.h5 --val_set ../../data/physionet_2012/val.h5
 #    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.MRNN --train_set ../../data/air_quality/train.h5 --val_set ../../data/air_quality/val.h5
-#    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.MRNN --train_set ../../data/ettm1/train.h5 --val_set ../../data/ettm1/val.h5 --test_set ../../data/ettm1/test.h5
+#    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.MRNN --train_set ../../data/ettm1/train.h5 --val_set ../../data/ettm1/val.h5
     codeDir: .
     gpuNum: 1
 

diff --git a/...OTS_tuning_configs/NonstationaryTransformer/NonstationaryTransformer_searching_config.yml b/...OTS_tuning_configs/NonstationaryTransformer/NonstationaryTransformer_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.NonstationaryTransformer --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.NonstationaryTransformer --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/benchmark_code/PyPOTS_tuning_configs/PatchTST/PatchTST_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/PatchTST/PatchTST_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.PatchTST --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.PatchTST --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/benchmark_code/PyPOTS_tuning_configs/Pyraformer/Pyraformer_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/Pyraformer/Pyraformer_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.Pyraformer --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.Pyraformer --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/...configs/SAITS/SAITS_AQI_tuning_space.json → .../SAITS/SAITS_BeijingAir_tuning_space.json b/...configs/SAITS/SAITS_AQI_tuning_space.json → .../SAITS/SAITS_BeijingAir_tuning_space.json
diff --git a/benchmark_code/PyPOTS_tuning_configs/SAITS/SAITS_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/SAITS/SAITS_searching_config.yml
@@ -3,8 +3,9 @@ authorName: WenjieDu
 trialConcurrency: 1
 trainingServicePlatform: local
 #searchSpacePath: SAITS_PhysioNet2012_tuning_space.json
-#searchSpacePath: SAITS_AQI_tuning_space.json
-searchSpacePath: SAITS_ETTh1_tuning_space.json
+#searchSpacePath: SAITS_BeijingAir_tuning_space.json
+#searchSpacePath: SAITS_ETTh1_tuning_space.json
+searchSpacePath: SAITS_Pedestrian_tuning_space.json
 multiThread: true
 useAnnotation: false
 tuner:
@@ -13,9 +14,10 @@ tuner:
 trial:
 #    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.SAITS --train_set ../../data/physionet_2012/train.h5 --val_set ../../data/physionet_2012/val.h5
 #    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.SAITS --train_set ../../data/air_quality/train.h5 --val_set ../../data/air_quality/val.h5
-    command:  pypots-cli tuning --model pypots.imputation.SAITS --train_set ../../data/ettm1/train.h5 --val_set ../../data/ettm1/val.h5
+#    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.SAITS --train_set ../../data/ettm1/train.h5 --val_set ../../data/ettm1/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.SAITS --train_set ../../data/melbourne_pedestrian_rate01_step24_point/train.h5 --val_set ../../data/melbourne_pedestrian_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/benchmark_code/PyPOTS_tuning_configs/SCINet/SCINet_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/SCINet/SCINet_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.SCINet --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.SCINet --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/benchmark_code/PyPOTS_tuning_configs/StemGNN/StemGNN_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/StemGNN/StemGNN_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.StemGNN --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.StemGNN --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true

diff --git a/...s/TimesNet/TimesNet_AQI_tuning_space.json → ...Net/TimesNet_BeijingAir_tuning_space.json b/...s/TimesNet/TimesNet_AQI_tuning_space.json → ...Net/TimesNet_BeijingAir_tuning_space.json
diff --git a/benchmark_code/PyPOTS_tuning_configs/TimesNet/TimesNet_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/TimesNet/TimesNet_searching_config.yml
@@ -3,7 +3,7 @@ authorName: WenjieDu
 trialConcurrency: 1
 trainingServicePlatform: local
 #searchSpacePath: TimesNet_PhysioNet2012_tuning_space.json
-#searchSpacePath: TimesNet_AQI_tuning_space.json
+#searchSpacePath: TimesNet_BeijingAir_tuning_space.json
 searchSpacePath: TimesNet_ETTh1_tuning_space.json
 multiThread: true
 useAnnotation: false

diff --git a/...sformer/Transformer_AQI_tuning_space.json → .../Transformer_BeijingAir_tuning_space.json b/...sformer/Transformer_AQI_tuning_space.json → .../Transformer_BeijingAir_tuning_space.json
diff --git a/benchmark_code/PyPOTS_tuning_configs/Transformer/Transformer_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/Transformer/Transformer_searching_config.yml
@@ -3,7 +3,7 @@ authorName: WenjieDu
 trialConcurrency: 1
 trainingServicePlatform: local
 #searchSpacePath: Transformer_PhysioNet2012_tuning_space.json
-#searchSpacePath: Transformer_AQI_tuning_space.json
+#searchSpacePath: Transformer_BeijingAir_tuning_space.json
 searchSpacePath: Transformer_ETTh1_tuning_space.json
 multiThread: true
 useAnnotation: false

diff --git a/...onfigs/US-GAN/USGAN_AQI_tuning_space.json → ...US-GAN/USGAN_BeijingAir_tuning_space.json b/...onfigs/US-GAN/USGAN_AQI_tuning_space.json → ...US-GAN/USGAN_BeijingAir_tuning_space.json
diff --git a/benchmark_code/PyPOTS_tuning_configs/US-GAN/USGAN_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/US-GAN/USGAN_searching_config.yml
@@ -3,17 +3,17 @@ authorName: WenjieDu
 trialConcurrency: 1
 trainingServicePlatform: local
 searchSpacePath: USGAN_PhysioNet2012_tuning_space.json
-#searchSpacePath: USGAN_AQI_tuning_space.json
+#searchSpacePath: USGAN_BeijingAir_tuning_space.json
 #searchSpacePath: USGAN_ETTh1_tuning_space.json
 multiThread: true
 useAnnotation: false
 tuner:
     builtinTunerName: Random
 
 trial:
-    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.USGAN --train_set ../../data/physionet_2012/train.h5 --val_set ../../data/physionet_2012/val.h5 --test_set ../../data/physionet_2012/test.h5
-#    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.USGAN --train_set ../../data/air_quality/train.h5 --val_set ../../data/air_quality/val.h5 --test_set ../../data/air_quality/test.h5
-#    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.USGAN --train_set ../../data/ettm1/train.h5 --val_set ../../data/ettm1/val.h5 --test_set ../../data/ettm1/test.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.USGAN --train_set ../../data/physionet_2012/train.h5 --val_set ../../data/physionet_2012/val.h5
+#    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.USGAN --train_set ../../data/air_quality/train.h5 --val_set ../../data/air_quality/val.h5
+#    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.USGAN --train_set ../../data/ettm1/train.h5 --val_set ../../data/ettm1/val.h5
     codeDir: .
     gpuNum: 1
 

diff --git a/benchmark_code/PyPOTS_tuning_configs/iTransformer/iTransformer_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/iTransformer/iTransformer_searching_config.yml
@@ -9,9 +9,9 @@ tuner:
     builtinTunerName: Random
 
 trial:
-    command:  pypots-cli tuning --model pypots.imputation.iTransformer --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
+    command: enable_tuning=1 pypots-cli tuning --model pypots.imputation.iTransformer --train_set ../../data/pems_traffic_rate01_step24_point/train.h5 --val_set ../../data/pems_traffic_rate01_step24_point/val.h5
     codeDir: .
-    gpuNum: 0
+    gpuNum: 1
 
 localConfig:
     useActiveGpu: true