From e58543b6f1844b617deab230e9a294f0b6e49664 Mon Sep 17 00:00:00 2001 From: Wenjie Du Date: Tue, 28 May 2024 16:20:04 +0800 Subject: [PATCH] docs: update tuning configs; --- .../Autoformer/Autoformer_searching_config.yml | 4 ++-- ...ng_space.json => CSDI_BeijingAir_tuning_space.json} | 0 .../CSDI/CSDI_searching_config.yml | 2 +- .../Crossformer/Crossformer_searching_config.yml | 4 ++-- .../DLinear/DLinear_searching_config.yml | 4 ++-- .../ETSformer/ETSformer_searching_config.yml | 4 ++-- .../FiLM/FiLM_searching_config.yml | 4 ++-- .../FreTS/FreTS_searching_config.yml | 4 ++-- .../GRUD/GRUD_searching_config.yml | 4 ++-- .../Informer/Informer_searching_config.yml | 4 ++-- .../Koopa/Koopa_searching_config.yml | 4 ++-- .../MICN/MICN_searching_config.yml | 4 ++-- ...ng_space.json => MRNN_BeijingAir_tuning_space.json} | 0 .../MRNN/MRNN_searching_config.yml | 6 +++--- .../NonstationaryTransformer_searching_config.yml | 4 ++-- .../PatchTST/PatchTST_searching_config.yml | 4 ++-- .../Pyraformer/Pyraformer_searching_config.yml | 4 ++-- ...g_space.json => SAITS_BeijingAir_tuning_space.json} | 0 .../SAITS/SAITS_searching_config.yml | 10 ++++++---- .../SCINet/SCINet_searching_config.yml | 4 ++-- .../StemGNN/StemGNN_searching_config.yml | 4 ++-- ...pace.json => TimesNet_BeijingAir_tuning_space.json} | 0 .../TimesNet/TimesNet_searching_config.yml | 2 +- ...e.json => Transformer_BeijingAir_tuning_space.json} | 0 .../Transformer/Transformer_searching_config.yml | 2 +- ...g_space.json => USGAN_BeijingAir_tuning_space.json} | 0 .../US-GAN/USGAN_searching_config.yml | 8 ++++---- .../iTransformer/iTransformer_searching_config.yml | 4 ++-- 28 files changed, 48 insertions(+), 46 deletions(-) rename benchmark_code/PyPOTS_tuning_configs/CSDI/{CSDI_AQI_tuning_space.json => CSDI_BeijingAir_tuning_space.json} (100%) rename benchmark_code/PyPOTS_tuning_configs/MRNN/{MRNN_AQI_tuning_space.json => MRNN_BeijingAir_tuning_space.json} (100%) rename benchmark_code/PyPOTS_tuning_configs/SAITS/{SAITS_AQI_tuning_space.json => SAITS_BeijingAir_tuning_space.json} (100%) rename benchmark_code/PyPOTS_tuning_configs/TimesNet/{TimesNet_AQI_tuning_space.json => TimesNet_BeijingAir_tuning_space.json} (100%) rename benchmark_code/PyPOTS_tuning_configs/Transformer/{Transformer_AQI_tuning_space.json => Transformer_BeijingAir_tuning_space.json} (100%) rename benchmark_code/PyPOTS_tuning_configs/US-GAN/{USGAN_AQI_tuning_space.json => USGAN_BeijingAir_tuning_space.json} (100%) diff --git a/benchmark_code/PyPOTS_tuning_configs/Autoformer/Autoformer_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/Autoformer/Autoformer_searching_config.yml index ffee3be..f8a9c92 100644 --- 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/benchmark_code/PyPOTS_tuning_configs/CSDI/CSDI_AQI_tuning_space.json b/benchmark_code/PyPOTS_tuning_configs/CSDI/CSDI_BeijingAir_tuning_space.json similarity index 100% rename from benchmark_code/PyPOTS_tuning_configs/CSDI/CSDI_AQI_tuning_space.json rename to benchmark_code/PyPOTS_tuning_configs/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 index aca8a3b..bef5301 100644 --- 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 index 85e1283..449aea1 100644 --- 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 index 5a2e857..5beddd2 100644 --- 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 index 5a2e857..5beddd2 100644 --- 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 index c98de9c..ed75537 100644 --- 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 index b29ac1e..80c2415 100644 --- 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 index 058b4ac..1ca01d0 100644 --- 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 index d7e9c3a..337c84d 100644 --- 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 index 1e4ab37..0629399 100644 --- 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 index f96c267..e334b0b 100644 --- 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/benchmark_code/PyPOTS_tuning_configs/MRNN/MRNN_AQI_tuning_space.json b/benchmark_code/PyPOTS_tuning_configs/MRNN/MRNN_BeijingAir_tuning_space.json similarity index 100% rename from benchmark_code/PyPOTS_tuning_configs/MRNN/MRNN_AQI_tuning_space.json rename to benchmark_code/PyPOTS_tuning_configs/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 index 31d07f3..a160dd6 100644 --- a/benchmark_code/PyPOTS_tuning_configs/MRNN/MRNN_searching_config.yml +++ b/benchmark_code/PyPOTS_tuning_configs/MRNN/MRNN_searching_config.yml @@ -3,7 +3,7 @@ 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 @@ -11,9 +11,9 @@ 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/benchmark_code/PyPOTS_tuning_configs/NonstationaryTransformer/NonstationaryTransformer_searching_config.yml b/benchmark_code/PyPOTS_tuning_configs/NonstationaryTransformer/NonstationaryTransformer_searching_config.yml index 698fcb1..f3f6f37 100644 --- a/benchmark_code/PyPOTS_tuning_configs/NonstationaryTransformer/NonstationaryTransformer_searching_config.yml +++ b/benchmark_code/PyPOTS_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 index 8422421..3f7d958 100644 --- 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 index b0a99df..36bb57f 100644 --- 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/benchmark_code/PyPOTS_tuning_configs/SAITS/SAITS_AQI_tuning_space.json b/benchmark_code/PyPOTS_tuning_configs/SAITS/SAITS_BeijingAir_tuning_space.json similarity index 100% rename from benchmark_code/PyPOTS_tuning_configs/SAITS/SAITS_AQI_tuning_space.json rename to benchmark_code/PyPOTS_tuning_configs/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 index d53714e..ae51f4f 100644 --- 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 index e90ca8c..9e7de63 100644 --- 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 index 77e68bd..e0e4fd6 100644 --- 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/benchmark_code/PyPOTS_tuning_configs/TimesNet/TimesNet_AQI_tuning_space.json b/benchmark_code/PyPOTS_tuning_configs/TimesNet/TimesNet_BeijingAir_tuning_space.json similarity index 100% rename from benchmark_code/PyPOTS_tuning_configs/TimesNet/TimesNet_AQI_tuning_space.json rename to benchmark_code/PyPOTS_tuning_configs/TimesNet/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 index 85106b4..f56f22d 100644 --- 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/benchmark_code/PyPOTS_tuning_configs/Transformer/Transformer_AQI_tuning_space.json b/benchmark_code/PyPOTS_tuning_configs/Transformer/Transformer_BeijingAir_tuning_space.json similarity index 100% rename from benchmark_code/PyPOTS_tuning_configs/Transformer/Transformer_AQI_tuning_space.json rename to benchmark_code/PyPOTS_tuning_configs/Transformer/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 index defe395..481737b 100644 --- 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/benchmark_code/PyPOTS_tuning_configs/US-GAN/USGAN_AQI_tuning_space.json b/benchmark_code/PyPOTS_tuning_configs/US-GAN/USGAN_BeijingAir_tuning_space.json similarity index 100% rename from benchmark_code/PyPOTS_tuning_configs/US-GAN/USGAN_AQI_tuning_space.json rename to benchmark_code/PyPOTS_tuning_configs/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 index 06f5573..a6e7b2e 100644 --- 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,7 +3,7 @@ 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 @@ -11,9 +11,9 @@ 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 index 45cf933..2a02e2a 100644 --- 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