first commit
This commit is contained in:
20
scripts/anomaly_detection/MSL/Autoformer.sh
Normal file
20
scripts/anomaly_detection/MSL/Autoformer.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/MSL \
|
||||
--model_id MSL \
|
||||
--model Autoformer \
|
||||
--data MSL \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 55 \
|
||||
--c_out 55 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
20
scripts/anomaly_detection/MSL/Crossformer.sh
Normal file
20
scripts/anomaly_detection/MSL/Crossformer.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/MSL \
|
||||
--model_id MSL \
|
||||
--model Crossformer \
|
||||
--data MSL \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 55 \
|
||||
--c_out 55 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
20
scripts/anomaly_detection/MSL/DLinear.sh
Normal file
20
scripts/anomaly_detection/MSL/DLinear.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/MSL \
|
||||
--model_id MSL \
|
||||
--model DLinear \
|
||||
--data MSL \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 100 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 55 \
|
||||
--c_out 55 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
21
scripts/anomaly_detection/MSL/ETSformer.sh
Normal file
21
scripts/anomaly_detection/MSL/ETSformer.sh
Normal file
@ -0,0 +1,21 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/MSL \
|
||||
--model_id MSL \
|
||||
--model ETSformer \
|
||||
--data MSL \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 100 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--d_layers 3 \
|
||||
--enc_in 55 \
|
||||
--c_out 55 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
20
scripts/anomaly_detection/MSL/FEDformer.sh
Normal file
20
scripts/anomaly_detection/MSL/FEDformer.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/MSL \
|
||||
--model_id MSL \
|
||||
--model FEDformer \
|
||||
--data MSL \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 55 \
|
||||
--c_out 55 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
20
scripts/anomaly_detection/MSL/FiLM.sh
Normal file
20
scripts/anomaly_detection/MSL/FiLM.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=6
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/MSL \
|
||||
--model_id MSL \
|
||||
--model FiLM \
|
||||
--data MSL \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 100 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 55 \
|
||||
--c_out 55 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 32 \
|
||||
--train_epochs 10
|
20
scripts/anomaly_detection/MSL/Informer.sh
Normal file
20
scripts/anomaly_detection/MSL/Informer.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/MSL \
|
||||
--model_id MSL \
|
||||
--model Informer \
|
||||
--data MSL \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 55 \
|
||||
--c_out 55 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
20
scripts/anomaly_detection/MSL/LightTS.sh
Normal file
20
scripts/anomaly_detection/MSL/LightTS.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/MSL \
|
||||
--model_id MSL \
|
||||
--model LightTS \
|
||||
--data MSL \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 55 \
|
||||
--c_out 55 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
20
scripts/anomaly_detection/MSL/MICN.sh
Normal file
20
scripts/anomaly_detection/MSL/MICN.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/MSL \
|
||||
--model_id MSL \
|
||||
--model MICN \
|
||||
--data MSL \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 55 \
|
||||
--c_out 55 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
20
scripts/anomaly_detection/MSL/Pyraformer.sh
Normal file
20
scripts/anomaly_detection/MSL/Pyraformer.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/MSL \
|
||||
--model_id MSL \
|
||||
--model Pyraformer \
|
||||
--data MSL \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 55 \
|
||||
--c_out 55 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
20
scripts/anomaly_detection/MSL/Reformer.sh
Normal file
20
scripts/anomaly_detection/MSL/Reformer.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/MSL \
|
||||
--model_id MSL \
|
||||
--model Reformer \
|
||||
--data MSL \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 55 \
|
||||
--c_out 55 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
21
scripts/anomaly_detection/MSL/TimesNet.sh
Normal file
21
scripts/anomaly_detection/MSL/TimesNet.sh
Normal file
@ -0,0 +1,21 @@
|
||||
export CUDA_VISIBLE_DEVICES=2
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/MSL \
|
||||
--model_id MSL \
|
||||
--model TimesNet \
|
||||
--data MSL \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 8 \
|
||||
--d_ff 16 \
|
||||
--e_layers 1 \
|
||||
--enc_in 55 \
|
||||
--c_out 55 \
|
||||
--top_k 3 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 1
|
20
scripts/anomaly_detection/MSL/Transformer.sh
Normal file
20
scripts/anomaly_detection/MSL/Transformer.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/MSL \
|
||||
--model_id MSL \
|
||||
--model Transformer \
|
||||
--data MSL \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 55 \
|
||||
--c_out 55 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
20
scripts/anomaly_detection/MSL/iTransformer.sh
Normal file
20
scripts/anomaly_detection/MSL/iTransformer.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/MSL \
|
||||
--model_id MSL \
|
||||
--model iTransformer \
|
||||
--data MSL \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 55 \
|
||||
--c_out 55 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
20
scripts/anomaly_detection/PSM/Autoformer.sh
Normal file
20
scripts/anomaly_detection/PSM/Autoformer.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=6
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PSM \
|
||||
--model_id PSM \
|
||||
--model Autoformer \
|
||||
--data PSM \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 25 \
|
||||
--c_out 25 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
20
scripts/anomaly_detection/PSM/DLinear.sh
Normal file
20
scripts/anomaly_detection/PSM/DLinear.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=6
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PSM \
|
||||
--model_id PSM \
|
||||
--model DLinear \
|
||||
--data PSM \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 100 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 25 \
|
||||
--c_out 25 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
21
scripts/anomaly_detection/PSM/TimesNet.sh
Normal file
21
scripts/anomaly_detection/PSM/TimesNet.sh
Normal file
@ -0,0 +1,21 @@
|
||||
export CUDA_VISIBLE_DEVICES=6
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PSM \
|
||||
--model_id PSM \
|
||||
--model TimesNet \
|
||||
--data PSM \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--e_layers 2 \
|
||||
--enc_in 25 \
|
||||
--c_out 25 \
|
||||
--top_k 3 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
20
scripts/anomaly_detection/PSM/Transformer.sh
Normal file
20
scripts/anomaly_detection/PSM/Transformer.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=6
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PSM \
|
||||
--model_id PSM \
|
||||
--model Transformer \
|
||||
--data PSM \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 25 \
|
||||
--c_out 25 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
20
scripts/anomaly_detection/SMAP/Autoformer.sh
Normal file
20
scripts/anomaly_detection/SMAP/Autoformer.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=7
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SMAP \
|
||||
--model_id SMAP \
|
||||
--model Autoformer \
|
||||
--data SMAP \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 25 \
|
||||
--c_out 25 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
21
scripts/anomaly_detection/SMAP/TimesNet.sh
Normal file
21
scripts/anomaly_detection/SMAP/TimesNet.sh
Normal file
@ -0,0 +1,21 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SMAP \
|
||||
--model_id SMAP \
|
||||
--model TimesNet \
|
||||
--data SMAP \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 25 \
|
||||
--c_out 25 \
|
||||
--top_k 3 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
20
scripts/anomaly_detection/SMAP/Transformer.sh
Normal file
20
scripts/anomaly_detection/SMAP/Transformer.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=7
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SMAP \
|
||||
--model_id SMAP \
|
||||
--model Transformer \
|
||||
--data SMAP \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 25 \
|
||||
--c_out 25 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
20
scripts/anomaly_detection/SMD/Autoformer.sh
Normal file
20
scripts/anomaly_detection/SMD/Autoformer.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=2
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SMD \
|
||||
--model_id SMD \
|
||||
--model Autoformer \
|
||||
--data SMD \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 38 \
|
||||
--c_out 38 \
|
||||
--anomaly_ratio 0.5 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
21
scripts/anomaly_detection/SMD/TimesNet.sh
Normal file
21
scripts/anomaly_detection/SMD/TimesNet.sh
Normal file
@ -0,0 +1,21 @@
|
||||
export CUDA_VISIBLE_DEVICES=2
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SMD \
|
||||
--model_id SMD \
|
||||
--model TimesNet \
|
||||
--data SMD \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--e_layers 2 \
|
||||
--enc_in 38 \
|
||||
--c_out 38 \
|
||||
--top_k 5 \
|
||||
--anomaly_ratio 0.5 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
20
scripts/anomaly_detection/SMD/Transformer.sh
Normal file
20
scripts/anomaly_detection/SMD/Transformer.sh
Normal file
@ -0,0 +1,20 @@
|
||||
export CUDA_VISIBLE_DEVICES=2
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SMD \
|
||||
--model_id SMD \
|
||||
--model Transformer \
|
||||
--data SMD \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 38 \
|
||||
--c_out 38 \
|
||||
--anomaly_ratio 0.5 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 10
|
21
scripts/anomaly_detection/SWAT/Autoformer.sh
Normal file
21
scripts/anomaly_detection/SWAT/Autoformer.sh
Normal file
@ -0,0 +1,21 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SWaT \
|
||||
--model_id SWAT \
|
||||
--model Autoformer \
|
||||
--data SWAT \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 51 \
|
||||
--c_out 51 \
|
||||
--top_k 3 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
161
scripts/anomaly_detection/SWAT/TimesNet.sh
Normal file
161
scripts/anomaly_detection/SWAT/TimesNet.sh
Normal file
@ -0,0 +1,161 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SWaT \
|
||||
--model_id SWAT \
|
||||
--model TimesNet \
|
||||
--data SWAT \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 8 \
|
||||
--d_ff 8 \
|
||||
--e_layers 3 \
|
||||
--enc_in 51 \
|
||||
--c_out 51 \
|
||||
--top_k 3 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SWaT \
|
||||
--model_id SWAT \
|
||||
--model TimesNet \
|
||||
--data SWAT \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 16 \
|
||||
--d_ff 16 \
|
||||
--e_layers 3 \
|
||||
--enc_in 51 \
|
||||
--c_out 51 \
|
||||
--top_k 3 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SWaT \
|
||||
--model_id SWAT \
|
||||
--model TimesNet \
|
||||
--data SWAT \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 32 \
|
||||
--d_ff 32 \
|
||||
--e_layers 3 \
|
||||
--enc_in 51 \
|
||||
--c_out 51 \
|
||||
--top_k 3 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SWaT \
|
||||
--model_id SWAT \
|
||||
--model TimesNet \
|
||||
--data SWAT \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--e_layers 3 \
|
||||
--enc_in 51 \
|
||||
--c_out 51 \
|
||||
--top_k 3 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SWaT \
|
||||
--model_id SWAT \
|
||||
--model TimesNet \
|
||||
--data SWAT \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 8 \
|
||||
--d_ff 8 \
|
||||
--e_layers 2 \
|
||||
--enc_in 51 \
|
||||
--c_out 51 \
|
||||
--top_k 3 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SWaT \
|
||||
--model_id SWAT \
|
||||
--model TimesNet \
|
||||
--data SWAT \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 16 \
|
||||
--d_ff 16 \
|
||||
--e_layers 2 \
|
||||
--enc_in 51 \
|
||||
--c_out 51 \
|
||||
--top_k 3 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SWaT \
|
||||
--model_id SWAT \
|
||||
--model TimesNet \
|
||||
--data SWAT \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 32 \
|
||||
--d_ff 32 \
|
||||
--e_layers 2 \
|
||||
--enc_in 51 \
|
||||
--c_out 51 \
|
||||
--top_k 3 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SWaT \
|
||||
--model_id SWAT \
|
||||
--model TimesNet \
|
||||
--data SWAT \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--e_layers 2 \
|
||||
--enc_in 51 \
|
||||
--c_out 51 \
|
||||
--top_k 3 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
21
scripts/anomaly_detection/SWAT/Transformer.sh
Normal file
21
scripts/anomaly_detection/SWAT/Transformer.sh
Normal file
@ -0,0 +1,21 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
python -u run.py \
|
||||
--task_name anomaly_detection \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SWaT \
|
||||
--model_id SWAT \
|
||||
--model Transformer \
|
||||
--data SWAT \
|
||||
--features M \
|
||||
--seq_len 100 \
|
||||
--pred_len 0 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--e_layers 3 \
|
||||
--enc_in 51 \
|
||||
--c_out 51 \
|
||||
--top_k 3 \
|
||||
--anomaly_ratio 1 \
|
||||
--batch_size 128 \
|
||||
--train_epochs 3
|
183
scripts/classification/Autoformer.sh
Normal file
183
scripts/classification/Autoformer.sh
Normal file
@ -0,0 +1,183 @@
|
||||
export CUDA_VISIBLE_DEVICES=5
|
||||
|
||||
model_name=Autoformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
183
scripts/classification/Crossformer.sh
Normal file
183
scripts/classification/Crossformer.sh
Normal file
@ -0,0 +1,183 @@
|
||||
export CUDA_VISIBLE_DEVICES=3
|
||||
|
||||
model_name=Crossformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
183
scripts/classification/DLinear.sh
Normal file
183
scripts/classification/DLinear.sh
Normal file
@ -0,0 +1,183 @@
|
||||
export CUDA_VISIBLE_DEVICES=5
|
||||
|
||||
model_name=DLinear
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
193
scripts/classification/ETSformer.sh
Normal file
193
scripts/classification/ETSformer.sh
Normal file
@ -0,0 +1,193 @@
|
||||
export CUDA_VISIBLE_DEVICES=5
|
||||
|
||||
model_name=ETSformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--d_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--d_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--d_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--d_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--d_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--d_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--d_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--d_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--d_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--d_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
183
scripts/classification/FEDformer.sh
Normal file
183
scripts/classification/FEDformer.sh
Normal file
@ -0,0 +1,183 @@
|
||||
export CUDA_VISIBLE_DEVICES=5
|
||||
|
||||
model_name=FEDformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
185
scripts/classification/FiLM.sh
Normal file
185
scripts/classification/FiLM.sh
Normal file
@ -0,0 +1,185 @@
|
||||
export CUDA_VISIBLE_DEVICES=7
|
||||
|
||||
model_name=FiLM
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--seq_len 1751 \
|
||||
--pred_len 1751 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
183
scripts/classification/Informer.sh
Normal file
183
scripts/classification/Informer.sh
Normal file
@ -0,0 +1,183 @@
|
||||
export CUDA_VISIBLE_DEVICES=5
|
||||
|
||||
model_name=Informer
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
183
scripts/classification/LightTS.sh
Normal file
183
scripts/classification/LightTS.sh
Normal file
@ -0,0 +1,183 @@
|
||||
export CUDA_VISIBLE_DEVICES=5
|
||||
|
||||
model_name=LightTS
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
184
scripts/classification/MICN.sh
Normal file
184
scripts/classification/MICN.sh
Normal file
@ -0,0 +1,184 @@
|
||||
export CUDA_VISIBLE_DEVICES=7
|
||||
|
||||
model_name=MICN
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--c_out 3 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
183
scripts/classification/PatchTST.sh
Normal file
183
scripts/classification/PatchTST.sh
Normal file
@ -0,0 +1,183 @@
|
||||
export CUDA_VISIBLE_DEVICES=3
|
||||
|
||||
model_name=PatchTST
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
183
scripts/classification/Pyraformer.sh
Normal file
183
scripts/classification/Pyraformer.sh
Normal file
@ -0,0 +1,183 @@
|
||||
export CUDA_VISIBLE_DEVICES=5
|
||||
|
||||
model_name=Pyraformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 4 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 4 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 4 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 4 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 4 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 4 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 4 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 4 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 4 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 4 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
183
scripts/classification/Reformer.sh
Normal file
183
scripts/classification/Reformer.sh
Normal file
@ -0,0 +1,183 @@
|
||||
export CUDA_VISIBLE_DEVICES=5
|
||||
|
||||
model_name=Reformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
182
scripts/classification/TimesNet.sh
Normal file
182
scripts/classification/TimesNet.sh
Normal file
@ -0,0 +1,182 @@
|
||||
export CUDA_VISIBLE_DEVICES=3
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model TimesNet \
|
||||
--data UEA \
|
||||
--e_layers 2 \
|
||||
--batch_size 16 \
|
||||
--d_model 16 \
|
||||
--d_ff 32 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 30 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model TimesNet \
|
||||
--data UEA \
|
||||
--e_layers 2 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--num_kernels 4 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 30 \
|
||||
--patience 10
|
||||
|
||||
python run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model TimesNet \
|
||||
--data UEA \
|
||||
--e_layers 2 \
|
||||
--batch_size 16 \
|
||||
--d_model 32 \
|
||||
--d_ff 64 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 30 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model TimesNet \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 16 \
|
||||
--d_ff 32 \
|
||||
--top_k 1 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 30 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model TimesNet \
|
||||
--data UEA \
|
||||
--e_layers 2 \
|
||||
--batch_size 16 \
|
||||
--d_model 16 \
|
||||
--d_ff 32 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 60 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model TimesNet \
|
||||
--data UEA \
|
||||
--e_layers 6 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 30 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model TimesNet \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 16 \
|
||||
--d_ff 32 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 30 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model TimesNet \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 32 \
|
||||
--d_ff 32 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 30 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model TimesNet \
|
||||
--data UEA \
|
||||
--e_layers 2 \
|
||||
--batch_size 16 \
|
||||
--d_model 32 \
|
||||
--d_ff 32 \
|
||||
--top_k 2 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 30 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model TimesNet \
|
||||
--data UEA \
|
||||
--e_layers 2 \
|
||||
--batch_size 16 \
|
||||
--d_model 32 \
|
||||
--d_ff 64 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 30 \
|
||||
--patience 10
|
183
scripts/classification/Transformer.sh
Normal file
183
scripts/classification/Transformer.sh
Normal file
@ -0,0 +1,183 @@
|
||||
export CUDA_VISIBLE_DEVICES=4
|
||||
|
||||
model_name=Transformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10
|
193
scripts/classification/iTransformer.sh
Normal file
193
scripts/classification/iTransformer.sh
Normal file
@ -0,0 +1,193 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=iTransformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 2048 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10 \
|
||||
--enc_in 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10 \
|
||||
--enc_in 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Handwriting/ \
|
||||
--model_id Handwriting \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10 \
|
||||
--enc_in 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/Heartbeat/ \
|
||||
--model_id Heartbeat \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10 \
|
||||
--enc_in 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/JapaneseVowels/ \
|
||||
--model_id JapaneseVowels \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10 \
|
||||
--enc_in 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/PEMS-SF/ \
|
||||
--model_id PEMS-SF \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10 \
|
||||
--enc_in 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP1/ \
|
||||
--model_id SelfRegulationSCP1 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10 \
|
||||
--enc_in 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SelfRegulationSCP2/ \
|
||||
--model_id SelfRegulationSCP2 \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10 \
|
||||
--enc_in 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/SpokenArabicDigits/ \
|
||||
--model_id SpokenArabicDigits \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10 \
|
||||
--enc_in 3
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/UWaveGestureLibrary/ \
|
||||
--model_id UWaveGestureLibrary \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10 \
|
||||
--enc_in 3
|
32
scripts/classification/xPatch_SparseChannel.sh
Normal file
32
scripts/classification/xPatch_SparseChannel.sh
Normal file
@ -0,0 +1,32 @@
|
||||
#!/bin/bash
|
||||
|
||||
# xPatch_SparseChannel Classification Training Script for FaceDetection Dataset
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=xPatch_SparseChannel
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/FaceDetection/ \
|
||||
--model_id FaceDetection \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 64 \
|
||||
--seq_len 62 \
|
||||
--enc_in 144 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--n_heads 8 \
|
||||
--patch_len 16 \
|
||||
--stride 8 \
|
||||
--moving_avg 25 \
|
||||
--dropout 0.1 \
|
||||
--des 'xPatch_SparseChannel_FaceDetection' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.0005 \
|
||||
--train_epochs 100 \
|
||||
--patience 5 \
|
||||
--revin 1 \
|
||||
--k_graph 8
|
89
scripts/exogenous_forecast/ECL/TimeXer.sh
Normal file
89
scripts/exogenous_forecast/ECL/TimeXer.sh
Normal file
@ -0,0 +1,89 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
model_name=TimeXer
|
||||
des='Timexer-MS'
|
||||
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des $des \
|
||||
--batch_size 4 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des $des \
|
||||
--batch_size 32 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_336 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des $des \
|
||||
--batch_size 32 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_720 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 3 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des $des \
|
||||
--d_model 512 \
|
||||
--itr 1
|
114
scripts/exogenous_forecast/EPF/TimeXer.sh
Normal file
114
scripts/exogenous_forecast/EPF/TimeXer.sh
Normal file
@ -0,0 +1,114 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=TimeXer
|
||||
des='Timexer-MS'
|
||||
patch_len=24
|
||||
|
||||
|
||||
python -u run.py \
|
||||
--is_training 1 \
|
||||
--task_name long_term_forecast \
|
||||
--root_path ./dataset/EPF/ \
|
||||
--data_path NP.csv \
|
||||
--model_id NP_168_24 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 168 \
|
||||
--pred_len 24 \
|
||||
--e_layers 3 \
|
||||
--enc_in 3 \
|
||||
--dec_in 3 \
|
||||
--c_out 1 \
|
||||
--des $des \
|
||||
--patch_len $patch_len \
|
||||
--d_model 512 \
|
||||
--d_ff 512 \
|
||||
--batch_size 4 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--is_training 1 \
|
||||
--task_name long_term_forecast \
|
||||
--root_path ./dataset/EPF/ \
|
||||
--data_path PJM.csv \
|
||||
--model_id PJM_168_24 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 168 \
|
||||
--pred_len 24 \
|
||||
--e_layers 3 \
|
||||
--enc_in 3 \
|
||||
--dec_in 3 \
|
||||
--c_out 1 \
|
||||
--des $des \
|
||||
--patch_len $patch_len \
|
||||
--d_model 512 \
|
||||
--batch_size 16 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--is_training 1 \
|
||||
--task_name long_term_forecast \
|
||||
--root_path ./dataset/EPF/ \
|
||||
--data_path BE.csv \
|
||||
--model_id BE_168_24 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 168 \
|
||||
--pred_len 24 \
|
||||
--e_layers 2 \
|
||||
--enc_in 3 \
|
||||
--dec_in 3 \
|
||||
--c_out 1 \
|
||||
--des $des \
|
||||
--patch_len $patch_len \
|
||||
--d_model 512 \
|
||||
--d_ff 512 \
|
||||
--batch_size 16 \
|
||||
--itr 1
|
||||
|
||||
|
||||
python -u run.py \
|
||||
--is_training 1 \
|
||||
--task_name long_term_forecast \
|
||||
--root_path ./dataset/EPF/ \
|
||||
--data_path FR.csv \
|
||||
--model_id FR_168_24 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 168 \
|
||||
--pred_len 24 \
|
||||
--e_layers 2 \
|
||||
--enc_in 3 \
|
||||
--dec_in 3 \
|
||||
--c_out 1 \
|
||||
--des $des \
|
||||
--patch_len $patch_len \
|
||||
--batch_size 16 \
|
||||
--d_model 512 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--is_training 1 \
|
||||
--task_name long_term_forecast \
|
||||
--root_path ./dataset/EPF/ \
|
||||
--data_path DE.csv \
|
||||
--model_id DE_168_24 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 168 \
|
||||
--pred_len 24 \
|
||||
--e_layers 1 \
|
||||
--enc_in 3 \
|
||||
--dec_in 3 \
|
||||
--c_out 1 \
|
||||
--des $des \
|
||||
--patch_len $patch_len \
|
||||
--batch_size 4 \
|
||||
--d_model 512 \
|
||||
--itr 1
|
94
scripts/exogenous_forecast/ETTh1/TimeXer.sh
Normal file
94
scripts/exogenous_forecast/ETTh1/TimeXer.sh
Normal file
@ -0,0 +1,94 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
model_name=TimeXer
|
||||
des='Timexer-MS'
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_96_96 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 512 \
|
||||
--d_ff 512 \
|
||||
--des $des \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_96_192 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 2 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--batch_size 4 \
|
||||
--des $des \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_96_336 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 2 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 512 \
|
||||
--d_ff 512 \
|
||||
--batch_size 32 \
|
||||
--des $des \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_96_720 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 2 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 512 \
|
||||
--batch_size 128 \
|
||||
--des $des \
|
||||
--itr 1
|
95
scripts/exogenous_forecast/ETTh2/TimeXer.sh
Normal file
95
scripts/exogenous_forecast/ETTh2/TimeXer.sh
Normal file
@ -0,0 +1,95 @@
|
||||
export CUDA_VISIBLE_DEVICES=3
|
||||
|
||||
model_name=TimeXer
|
||||
des='Timexer-MS'
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_96_96 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--batch_size 128 \
|
||||
--des $des \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_96_192 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 128 \
|
||||
--d_ff 512 \
|
||||
--batch_size 128 \
|
||||
--des $des \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_96_336 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 2 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--batch_size 16 \
|
||||
--des $des \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_96_720 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 256 \
|
||||
--d_ff 512 \
|
||||
--des $des \
|
||||
--itr 1
|
92
scripts/exogenous_forecast/ETTm1/TimeXer.sh
Normal file
92
scripts/exogenous_forecast/ETTm1/TimeXer.sh
Normal file
@ -0,0 +1,92 @@
|
||||
export CUDA_VISIBLE_DEVICES=2
|
||||
|
||||
model_name=TimeXer
|
||||
des='Timexer-MS'
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_96_96 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 256 \
|
||||
--batch_size 128 \
|
||||
--des $des \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_96_192 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 128 \
|
||||
--batch_size 128 \
|
||||
--des $des \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_96_336 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 128 \
|
||||
--batch_size 128 \
|
||||
--des $des \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_96_720 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 128 \
|
||||
--batch_size 128 \
|
||||
--des $des \
|
||||
--itr 1
|
92
scripts/exogenous_forecast/ETTm2/TimeXer.sh
Normal file
92
scripts/exogenous_forecast/ETTm2/TimeXer.sh
Normal file
@ -0,0 +1,92 @@
|
||||
export CUDA_VISIBLE_DEVICES=2
|
||||
|
||||
model_name=TimeXer
|
||||
des='Timexer-MS'
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_96_96 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 512 \
|
||||
--batch_size 16 \
|
||||
--des $des \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_96_192 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 256 \
|
||||
--batch_size 4 \
|
||||
--des $des \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_96_336 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 128 \
|
||||
--batch_size 128 \
|
||||
--des $des \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_96_720 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--d_model 128 \
|
||||
--batch_size 128 \
|
||||
--des $des \
|
||||
--itr 1
|
96
scripts/exogenous_forecast/Traffic/TimeXer.sh
Normal file
96
scripts/exogenous_forecast/Traffic/TimeXer.sh
Normal file
@ -0,0 +1,96 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=TimeXer
|
||||
des='Timexer-MS'
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/traffic/ \
|
||||
--data_path traffic.csv \
|
||||
--model_id traffic_96_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 1 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 862 \
|
||||
--dec_in 862 \
|
||||
--c_out 862 \
|
||||
--d_model 512 \
|
||||
--des $des \
|
||||
--batch_size 4 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/traffic/ \
|
||||
--data_path traffic.csv \
|
||||
--model_id traffic_96_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 1 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 862 \
|
||||
--dec_in 862 \
|
||||
--c_out 862 \
|
||||
--d_model 512 \
|
||||
--des 'Exp' \
|
||||
--batch_size 4 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/traffic/ \
|
||||
--data_path traffic.csv \
|
||||
--model_id traffic_96_336 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 1 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 862 \
|
||||
--dec_in 862 \
|
||||
--c_out 862 \
|
||||
--d_model 512 \
|
||||
--des $des \
|
||||
--batch_size 4 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/traffic/ \
|
||||
--data_path traffic.csv \
|
||||
--model_id traffic_96_720 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 1 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 862 \
|
||||
--dec_in 862 \
|
||||
--c_out 862 \
|
||||
--d_model 512 \
|
||||
--des $des \
|
||||
--batch_size 4 \
|
||||
--itr 1
|
89
scripts/exogenous_forecast/Weather/TimeXer.sh
Normal file
89
scripts/exogenous_forecast/Weather/TimeXer.sh
Normal file
@ -0,0 +1,89 @@
|
||||
export CUDA_VISIBLE_DEVICES=3
|
||||
|
||||
model_name=TimeXer
|
||||
des='Timexer-MS'
|
||||
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_96_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--des $des \
|
||||
--d_model 128 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_96_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--des $des \
|
||||
--d_model 128 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_96_336 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--des $des \
|
||||
--d_model 128 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_96_720 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features MS \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--des $des \
|
||||
--d_model 128 \
|
||||
--itr 1
|
111
scripts/imputation/ECL_script/Autoformer.sh
Normal file
111
scripts/imputation/ECL_script/Autoformer.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Autoformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ECL_script/DLinear.sh
Normal file
111
scripts/imputation/ECL_script/DLinear.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=DLinear
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ECL_script/ETSformer.sh
Normal file
111
scripts/imputation/ECL_script/ETSformer.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=ETSformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 2 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 2 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 2 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 2 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ECL_script/FEDformer.sh
Normal file
111
scripts/imputation/ECL_script/FEDformer.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=FEDformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ECL_script/Informer.sh
Normal file
111
scripts/imputation/ECL_script/Informer.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Informer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ECL_script/LightTS.sh
Normal file
111
scripts/imputation/ECL_script/LightTS.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=LightTS
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ECL_script/Pyraformer.sh
Normal file
111
scripts/imputation/ECL_script/Pyraformer.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Pyraformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ECL_script/Reformer.sh
Normal file
111
scripts/imputation/ECL_script/Reformer.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Reformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ECL_script/TimesNet.sh
Normal file
111
scripts/imputation/ECL_script/TimesNet.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=7
|
||||
|
||||
model_name=TimesNet
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ECL_script/Transformer.sh
Normal file
111
scripts/imputation/ECL_script/Transformer.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Transformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ECL_script/iTransformer.sh
Normal file
111
scripts/imputation/ECL_script/iTransformer.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=iTransformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/Autoformer_ETTh1.sh
Normal file
111
scripts/imputation/ETT_script/Autoformer_ETTh1.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Autoformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/Autoformer_ETTh2.sh
Normal file
111
scripts/imputation/ETT_script/Autoformer_ETTh2.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Autoformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/Autoformer_ETTm1.sh
Normal file
111
scripts/imputation/ETT_script/Autoformer_ETTm1.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Autoformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/Autoformer_ETTm2.sh
Normal file
111
scripts/imputation/ETT_script/Autoformer_ETTm2.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Autoformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/Crossformer_ETTh1.sh
Normal file
111
scripts/imputation/ETT_script/Crossformer_ETTh1.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Crossformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/DLinear_ETTh1.sh
Normal file
111
scripts/imputation/ETT_script/DLinear_ETTh1.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=DLinear
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/FiLM_ETTh1.sh
Normal file
111
scripts/imputation/ETT_script/FiLM_ETTh1.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=6
|
||||
|
||||
model_name=FiLM
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
115
scripts/imputation/ETT_script/MICN_ETTh1.sh
Normal file
115
scripts/imputation/ETT_script/MICN_ETTh1.sh
Normal file
@ -0,0 +1,115 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=MICN
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--conv_kernel 12 16 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--conv_kernel 12 16 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--conv_kernel 12 16 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--conv_kernel 12 16 \
|
||||
--learning_rate 0.001
|
119
scripts/imputation/ETT_script/Nonstationary_Transformer_ETTh1.sh
Normal file
119
scripts/imputation/ETT_script/Nonstationary_Transformer_ETTh1.sh
Normal file
@ -0,0 +1,119 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Nonstationary_Transformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001 \
|
||||
--p_hidden_dims 256 256 \
|
||||
--p_hidden_layers 2
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001 \
|
||||
--p_hidden_dims 256 256 \
|
||||
--p_hidden_layers 2
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001 \
|
||||
--p_hidden_dims 256 256 \
|
||||
--p_hidden_layers 2
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001 \
|
||||
--p_hidden_dims 256 256 \
|
||||
--p_hidden_layers 2
|
110
scripts/imputation/ETT_script/TiDE_ETTh1.sh
Normal file
110
scripts/imputation/ETT_script/TiDE_ETTh1.sh
Normal file
@ -0,0 +1,110 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=TiDE
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 2 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 256 \
|
||||
--d_ff 256 \
|
||||
--dropout 0.3 \
|
||||
--learning_rate 0.1 \
|
||||
--patience 5 \
|
||||
--train_epochs 10 \
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 16 \
|
||||
--d_ff 32 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 16 \
|
||||
--d_ff 32 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 16 \
|
||||
--d_ff 32 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/TimesNet_ETTh1.sh
Normal file
111
scripts/imputation/ETT_script/TimesNet_ETTh1.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=TimesNet
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 16 \
|
||||
--d_ff 32 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 16 \
|
||||
--d_ff 32 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 16 \
|
||||
--d_ff 32 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 16 \
|
||||
--d_ff 32 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/TimesNet_ETTh2.sh
Normal file
111
scripts/imputation/ETT_script/TimesNet_ETTh2.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
model_name=TimesNet
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 32 \
|
||||
--d_ff 32 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 32 \
|
||||
--d_ff 32 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 32 \
|
||||
--d_ff 32 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 32 \
|
||||
--d_ff 32 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/TimesNet_ETTm1.sh
Normal file
111
scripts/imputation/ETT_script/TimesNet_ETTm1.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=2
|
||||
|
||||
model_name=TimesNet
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/TimesNet_ETTm2.sh
Normal file
111
scripts/imputation/ETT_script/TimesNet_ETTm2.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=5
|
||||
|
||||
model_name=TimesNet
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/Transformer_ETTh1.sh
Normal file
111
scripts/imputation/ETT_script/Transformer_ETTh1.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Transformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh1.csv \
|
||||
--model_id ETTh1_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTh1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/Transformer_ETTh2.sh
Normal file
111
scripts/imputation/ETT_script/Transformer_ETTh2.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Transformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/Transformer_ETTm1.sh
Normal file
111
scripts/imputation/ETT_script/Transformer_ETTm1.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Transformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm1.csv \
|
||||
--model_id ETTm1_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTm1 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/Transformer_ETTm2.sh
Normal file
111
scripts/imputation/ETT_script/Transformer_ETTm2.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Transformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTm2.csv \
|
||||
--model_id ETTm2_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTm2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/ETT_script/iTransformer_ETTh2.sh
Normal file
111
scripts/imputation/ETT_script/iTransformer_ETTh2.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=iTransformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/ETT-small/ \
|
||||
--data_path ETTh2.csv \
|
||||
--model_id ETTh2_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data ETTh2 \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 7 \
|
||||
--dec_in 7 \
|
||||
--c_out 7 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/Weather_script/Autoformer.sh
Normal file
111
scripts/imputation/Weather_script/Autoformer.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Autoformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/Weather_script/TimesNet.sh
Normal file
111
scripts/imputation/Weather_script/TimesNet.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=6
|
||||
|
||||
model_name=TimesNet
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--batch_size 16 \
|
||||
--d_model 64 \
|
||||
--d_ff 64 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 3 \
|
||||
--learning_rate 0.001
|
111
scripts/imputation/Weather_script/Transformer.sh
Normal file
111
scripts/imputation/Weather_script/Transformer.sh
Normal file
@ -0,0 +1,111 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Transformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_mask_0.125 \
|
||||
--mask_rate 0.125 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_mask_0.25 \
|
||||
--mask_rate 0.25 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_mask_0.375 \
|
||||
--mask_rate 0.375 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
||||
|
||||
python -u run.py \
|
||||
--task_name imputation \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/weather/ \
|
||||
--data_path weather.csv \
|
||||
--model_id weather_mask_0.5 \
|
||||
--mask_rate 0.5 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 0 \
|
||||
--pred_len 0 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 21 \
|
||||
--dec_in 21 \
|
||||
--c_out 21 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--top_k 5 \
|
||||
--learning_rate 0.001
|
@ -0,0 +1,28 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=PatchTST
|
||||
|
||||
for aug in jitter scaling permutation magwarp timewarp windowslice windowwarp rotation spawner dtwwarp shapedtwwarp wdba discdtw discsdtw
|
||||
do
|
||||
echo using augmentation: ${aug}
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10 \
|
||||
--augmentation_ratio 1 \
|
||||
--${aug}
|
||||
done
|
@ -0,0 +1,33 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=PatchTST
|
||||
for aug in jitter scaling permutation magwarp timewarp windowslice windowwarp rotation spawner dtwwarp shapedtwwarp discdtw discsdtw
|
||||
do
|
||||
for pred_len in 96 192 336 720
|
||||
do
|
||||
echo using augmentation: ${aug}
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/exchange_rate/ \
|
||||
--data_path exchange_rate.csv \
|
||||
--model_id Exchange_96_${pred_len} \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len ${pred_len} \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 8 \
|
||||
--dec_in 8 \
|
||||
--c_out 8 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--augmentation_ratio 1 \
|
||||
--${aug}
|
||||
done
|
||||
done
|
97
scripts/long_term_forecast/AugmentSample/ReadMe.md
Normal file
97
scripts/long_term_forecast/AugmentSample/ReadMe.md
Normal file
@ -0,0 +1,97 @@
|
||||
# Augmentation Feature Roadbook
|
||||
|
||||
Hi there! For those who are interested in testing
|
||||
augmentation techniques in `Time-Series-Library`.
|
||||
|
||||
For now, we have embedded several augmentation methods
|
||||
in this repo. We are still collecting publicly available
|
||||
augmentation algorithms, and we appreciate your valuable
|
||||
advice!
|
||||
|
||||
```
|
||||
The Implemented Augmentation Methods
|
||||
1. jitter
|
||||
2. scaling
|
||||
3. permutation
|
||||
4. magwarp
|
||||
5. timewarp
|
||||
6. windowslice
|
||||
7. windowwarp
|
||||
8. rotation
|
||||
9. spawner
|
||||
10. dtwwarp
|
||||
11. shapedtwwarp
|
||||
12. wdba (Specially Designed for Classification tasks)
|
||||
13. discdtw
|
||||
```
|
||||
|
||||
## Usage
|
||||
|
||||
In this folder, we present two sample of shell scripts
|
||||
doing augmentation in `Forecasting` and `Classification`
|
||||
tasks.
|
||||
|
||||
Take `Forecasting` task for example, we test multiple
|
||||
augmentation algorithms on `EthanolConcentration` dataset
|
||||
(a subset of the popular classification benchmark `UEA`)
|
||||
using `PatchTST` model.
|
||||
|
||||
```shell
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=PatchTST
|
||||
|
||||
for aug in jitter scaling permutation magwarp timewarp windowslice windowwarp rotation spawner dtwwarp shapedtwwarp wdba discdtw discsdtw
|
||||
do
|
||||
echo using augmentation: ${aug}
|
||||
|
||||
python -u run.py \
|
||||
--task_name classification \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/EthanolConcentration/ \
|
||||
--model_id EthanolConcentration \
|
||||
--model $model_name \
|
||||
--data UEA \
|
||||
--e_layers 3 \
|
||||
--batch_size 16 \
|
||||
--d_model 128 \
|
||||
--d_ff 256 \
|
||||
--top_k 3 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--learning_rate 0.001 \
|
||||
--train_epochs 100 \
|
||||
--patience 10 \
|
||||
--augmentation_ratio 1 \
|
||||
--${aug}
|
||||
done
|
||||
```
|
||||
|
||||
Here, parameter `augmentation_ratio` represents how many
|
||||
times do we want to perform our augmentation method.
|
||||
Parameter `${aug}` represents a string of augmentation
|
||||
type label.
|
||||
|
||||
The example here only perform augmentation once, so we
|
||||
can set `augmentation_ratio` to `1`, followed by one
|
||||
augmentation type label. Trivially, you can set
|
||||
`augmentation_ratio` to an integer `num` followed by
|
||||
`num` augmentation type labels.
|
||||
|
||||
The augmentation code obeys the same prototype of
|
||||
`Time-Series-Library`. If you want to adjust other
|
||||
training parameters, feel free to add arguments to the
|
||||
shell scripts and play around. The full list of parameters
|
||||
can be seen in `run.py`.
|
||||
|
||||
## Contact Us!
|
||||
|
||||
This piece of code is written and maintained by
|
||||
[Yunzhong Qiu](https://github.com/DigitalLifeYZQiu).
|
||||
We thank [Haixu Wu](https://github.com/wuhaixu2016) and
|
||||
[Jiaxiang Dong](https://github.com/dongjiaxiang) for
|
||||
insightful discussion and solid support.
|
||||
|
||||
If you have difficulties or find bugs in our code, please
|
||||
contact us:
|
||||
- Email: qiuyz24@mails.tsinghua.edu.cn
|
87
scripts/long_term_forecast/ECL_script/Autoformer.sh
Normal file
87
scripts/long_term_forecast/ECL_script/Autoformer.sh
Normal file
@ -0,0 +1,87 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
model_name=Autoformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_336 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_720 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
103
scripts/long_term_forecast/ECL_script/Crossformer.sh
Normal file
103
scripts/long_term_forecast/ECL_script/Crossformer.sh
Normal file
@ -0,0 +1,103 @@
|
||||
export CUDA_VISIBLE_DEVICES=7
|
||||
|
||||
model_name=Crossformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--d_model 256 \
|
||||
--d_ff 512 \
|
||||
--top_k 5 \
|
||||
--des 'Exp' \
|
||||
--batch_size 16 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--d_model 256 \
|
||||
--d_ff 512 \
|
||||
--top_k 5 \
|
||||
--des 'Exp' \
|
||||
--batch_size 16 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_336 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--d_model 256 \
|
||||
--d_ff 512 \
|
||||
--top_k 5 \
|
||||
--des 'Exp' \
|
||||
--batch_size 16 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_720 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--d_model 256 \
|
||||
--d_ff 512 \
|
||||
--top_k 5 \
|
||||
--des 'Exp' \
|
||||
--batch_size 16 \
|
||||
--itr 1
|
87
scripts/long_term_forecast/ECL_script/DLinear.sh
Normal file
87
scripts/long_term_forecast/ECL_script/DLinear.sh
Normal file
@ -0,0 +1,87 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
model_name=DLinear
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_336 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_720 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
87
scripts/long_term_forecast/ECL_script/ETSformer.sh
Normal file
87
scripts/long_term_forecast/ECL_script/ETSformer.sh
Normal file
@ -0,0 +1,87 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
model_name=ETSformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 2 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 2 \
|
||||
--d_layers 2 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_336 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 2 \
|
||||
--d_layers 2 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_720 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 2 \
|
||||
--d_layers 2 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
87
scripts/long_term_forecast/ECL_script/FEDformer.sh
Normal file
87
scripts/long_term_forecast/ECL_script/FEDformer.sh
Normal file
@ -0,0 +1,87 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
model_name=FEDformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_336 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_720 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
91
scripts/long_term_forecast/ECL_script/FiLM.sh
Normal file
91
scripts/long_term_forecast/ECL_script/FiLM.sh
Normal file
@ -0,0 +1,91 @@
|
||||
export CUDA_VISIBLE_DEVICES=3
|
||||
|
||||
model_name=FiLM
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features S \
|
||||
--seq_len 192 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--batch_size 4 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features S \
|
||||
--seq_len 192 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--batch_size 4 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_336 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features S \
|
||||
--seq_len 192 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--batch_size 4 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_720 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features S \
|
||||
--seq_len 192 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--batch_size 4 \
|
||||
--itr 1
|
87
scripts/long_term_forecast/ECL_script/Informer.sh
Normal file
87
scripts/long_term_forecast/ECL_script/Informer.sh
Normal file
@ -0,0 +1,87 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
model_name=Informer
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_336 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_720 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
87
scripts/long_term_forecast/ECL_script/Koopa.sh
Normal file
87
scripts/long_term_forecast/ECL_script/Koopa.sh
Normal file
@ -0,0 +1,87 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
model_name=Koopa
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_48 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--pred_len 48 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--learning_rate 0.001 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_192_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 192 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--learning_rate 0.001 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_288_144 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 288 \
|
||||
--pred_len 144 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--learning_rate 0.001 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_384_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 384 \
|
||||
--pred_len 192 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--learning_rate 0.001 \
|
||||
--itr 1
|
87
scripts/long_term_forecast/ECL_script/LightTS.sh
Normal file
87
scripts/long_term_forecast/ECL_script/LightTS.sh
Normal file
@ -0,0 +1,87 @@
|
||||
export CUDA_VISIBLE_DEVICES=1
|
||||
|
||||
model_name=LightTS
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_336 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_720 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
99
scripts/long_term_forecast/ECL_script/MICN.sh
Normal file
99
scripts/long_term_forecast/ECL_script/MICN.sh
Normal file
@ -0,0 +1,99 @@
|
||||
export CUDA_VISIBLE_DEVICES=4
|
||||
|
||||
model_name=MICN
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 96 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--d_model 256 \
|
||||
--d_ff 512 \
|
||||
--top_k 5 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 96 \
|
||||
--pred_len 192 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--d_model 256 \
|
||||
--d_ff 512 \
|
||||
--top_k 5 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_336 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 96 \
|
||||
--pred_len 336 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--d_model 256 \
|
||||
--d_ff 512 \
|
||||
--top_k 5 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_720 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 96 \
|
||||
--pred_len 720 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--d_model 256 \
|
||||
--d_ff 512 \
|
||||
--top_k 5 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
30
scripts/long_term_forecast/ECL_script/Mamba.sh
Normal file
30
scripts/long_term_forecast/ECL_script/Mamba.sh
Normal file
@ -0,0 +1,30 @@
|
||||
model_name=Mamba
|
||||
|
||||
for pred_len in 96 192 336 720
|
||||
# for pred_len in 336 720
|
||||
do
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_$pred_len'_'$pred_len \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len $pred_len \
|
||||
--label_len 48 \
|
||||
--pred_len $pred_len \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--enc_in 321 \
|
||||
--expand 2 \
|
||||
--d_ff 16 \
|
||||
--d_conv 4 \
|
||||
--c_out 321 \
|
||||
--d_model 128 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
|
||||
done
|
98
scripts/long_term_forecast/ECL_script/MultiPatchFormer.sh
Normal file
98
scripts/long_term_forecast/ECL_script/MultiPatchFormer.sh
Normal file
@ -0,0 +1,98 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=MultiPatchFormer
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 1 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--d_model 256 \
|
||||
--d_ff 512 \
|
||||
--des 'Exp' \
|
||||
--n_heads 8 \
|
||||
--batch_size 32 \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--d_model 256 \
|
||||
--d_ff 512 \
|
||||
--top_k 5 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_336 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--d_model 256 \
|
||||
--d_ff 512 \
|
||||
--top_k 5 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_720 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--d_model 256 \
|
||||
--d_ff 512 \
|
||||
--top_k 5 \
|
||||
--des 'Exp' \
|
||||
--itr 1
|
@ -0,0 +1,99 @@
|
||||
export CUDA_VISIBLE_DEVICES=0
|
||||
|
||||
model_name=Nonstationary_Transformer
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_96 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 96 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--p_hidden_dims 256 256 \
|
||||
--p_hidden_layers 2 \
|
||||
--d_model 2048
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_192 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 192 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--p_hidden_dims 256 256 \
|
||||
--p_hidden_layers 2 \
|
||||
--d_model 2048
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_336 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 336 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--p_hidden_dims 256 256 \
|
||||
--p_hidden_layers 2 \
|
||||
--d_model 2048
|
||||
|
||||
python -u run.py \
|
||||
--task_name long_term_forecast \
|
||||
--is_training 1 \
|
||||
--root_path ./dataset/electricity/ \
|
||||
--data_path electricity.csv \
|
||||
--model_id ECL_96_720 \
|
||||
--model $model_name \
|
||||
--data custom \
|
||||
--features M \
|
||||
--seq_len 96 \
|
||||
--label_len 48 \
|
||||
--pred_len 720 \
|
||||
--e_layers 2 \
|
||||
--d_layers 1 \
|
||||
--factor 3 \
|
||||
--enc_in 321 \
|
||||
--dec_in 321 \
|
||||
--c_out 321 \
|
||||
--des 'Exp' \
|
||||
--itr 1 \
|
||||
--p_hidden_dims 256 256 \
|
||||
--p_hidden_layers 2 \
|
||||
--d_model 2048
|
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user