first commit
This commit is contained in:
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
|
Reference in New Issue
Block a user