Commit 9c0053b7 authored by chenzk's avatar chenzk
Browse files

v1.0

parents
Pipeline #953 canceled with stages
export CUDA_VISIBLE_DEVICES=0
model_name=Flashformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather_96_96 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 48 \
--pred_len 96 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--itr 1 \
--batch_size 128 \
--train_epochs 3
python -u run.py \
--is_training 1 \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather_96_192 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 48 \
--pred_len 192 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--batch_size 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather_96_336 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 48 \
--pred_len 336 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--batch_size 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather_96_720 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 48 \
--pred_len 720 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--batch_size 128 \
--itr 1
model_name=iFlashformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather_96_96 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 48 \
--pred_len 96 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--batch_size 128 \
--itr 1 \
--train_epochs 3
python -u run.py \
--is_training 1 \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather_96_192 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 48 \
--pred_len 192 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--batch_size 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather_96_336 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 48 \
--pred_len 336 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--batch_size 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather_96_720 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 48 \
--pred_len 720 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--batch_size 128 \
--itr 1
\ No newline at end of file
export CUDA_VISIBLE_DEVICES=0
model_name=iTransformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_96_96 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 96 \
--e_layers 3 \
--enc_in 321 \
--dec_in 321 \
--c_out 321 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--batch_size 16 \
--learning_rate 0.0005 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_96_192 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 192 \
--e_layers 3 \
--enc_in 321 \
--dec_in 321 \
--c_out 321 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--batch_size 16 \
--learning_rate 0.0005 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_96_336 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 336 \
--e_layers 3 \
--enc_in 321 \
--dec_in 321 \
--c_out 321 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--batch_size 16 \
--learning_rate 0.0005 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_96_720 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 720 \
--e_layers 3 \
--enc_in 321 \
--dec_in 321 \
--c_out 321 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--batch_size 16 \
--learning_rate 0.0005 \
--itr 1
\ No newline at end of file
export CUDA_VISIBLE_DEVICES=1
model_name=iTransformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTh1.csv \
--model_id ETTh1_96_96 \
--model $model_name \
--data ETTh1 \
--features M \
--seq_len 96 \
--pred_len 96 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 256 \
--d_ff 256 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTh1.csv \
--model_id ETTh1_96_192 \
--model $model_name \
--data ETTh1 \
--features M \
--seq_len 96 \
--pred_len 192 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 256 \
--d_ff 256 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTh1.csv \
--model_id ETTh1_96_336 \
--model $model_name \
--data ETTh1 \
--features M \
--seq_len 96 \
--pred_len 336 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTh1.csv \
--model_id ETTh1_96_720 \
--model $model_name \
--data ETTh1 \
--features M \
--seq_len 96 \
--pred_len 720 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--itr 1
\ No newline at end of file
export CUDA_VISIBLE_DEVICES=1
model_name=iTransformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTh2.csv \
--model_id ETTh2_96_96 \
--model $model_name \
--data ETTh2 \
--features M \
--seq_len 96 \
--pred_len 96 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTh2.csv \
--model_id ETTh2_96_192 \
--model $model_name \
--data ETTh2 \
--features M \
--seq_len 96 \
--pred_len 192 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTh2.csv \
--model_id ETTh2_96_336 \
--model $model_name \
--data ETTh2 \
--features M \
--seq_len 96 \
--pred_len 336 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTh2.csv \
--model_id ETTh2_96_720 \
--model $model_name \
--data ETTh2 \
--features M \
--seq_len 96 \
--pred_len 720 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
\ No newline at end of file
export CUDA_VISIBLE_DEVICES=2
model_name=iTransformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTm1.csv \
--model_id ETTm1_96_96 \
--model $model_name \
--data ETTm1 \
--features M \
--seq_len 96 \
--pred_len 96 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTm1.csv \
--model_id ETTm1_96_192 \
--model $model_name \
--data ETTm1 \
--features M \
--seq_len 96 \
--pred_len 192 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTm1.csv \
--model_id ETTm1_96_336 \
--model $model_name \
--data ETTm1 \
--features M \
--seq_len 96 \
--pred_len 336 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTm1.csv \
--model_id ETTm1_96_720 \
--model $model_name \
--data ETTm1 \
--features M \
--seq_len 96 \
--pred_len 720 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
\ No newline at end of file
export CUDA_VISIBLE_DEVICES==0
model_name=iTransformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2_96_96 \
--model $model_name \
--data ETTm2 \
--features M \
--seq_len 96 \
--pred_len 96 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2_96_192 \
--model $model_name \
--data ETTm2 \
--features M \
--seq_len 96 \
--pred_len 192 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2_96_336 \
--model $model_name \
--data ETTm2 \
--features M \
--seq_len 96 \
--pred_len 336 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2_96_720 \
--model $model_name \
--data ETTm2 \
--features M \
--seq_len 96 \
--pred_len 720 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
export CUDA_VISIBLE_DEVICES==0
model_name=iTransformer
# 默认按天预测
python -u run.py \
--is_training 0 \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2_96_96 \
--model $model_name \
--data ETTm2 \
--features M \
--freq '15min' \
--seq_len 96 \
--pred_len 96 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
export CUDA_VISIBLE_DEVICES==0
model_name=iTransformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2_96_96 \
--model $model_name \
--data ETTm2 \
--train_epochs 10 \
--features M \
--freq '15min' \
--seq_len 96 \
--pred_len 96 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2_96_192 \
--model $model_name \
--data ETTm2 \
--train_epochs 10 \
--features M \
--freq '15min' \
--seq_len 96 \
--pred_len 192 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2_96_336 \
--model $model_name \
--data ETTm2 \
--train_epochs 10 \
--features M \
--freq '15min' \
--seq_len 96 \
--pred_len 336 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2_96_720 \
--model $model_name \
--data ETTm2 \
--train_epochs 10 \
--features M \
--freq '15min' \
--seq_len 96 \
--pred_len 720 \
--e_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
export CUDA_VISIBLE_DEVICES=3
model_name=iTransformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/exchange_rate/ \
--data_path exchange_rate.csv \
--model_id Exchange_96_96 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 96 \
--e_layers 2 \
--enc_in 8 \
--dec_in 8 \
--c_out 8 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/exchange_rate/ \
--data_path exchange_rate.csv \
--model_id Exchange_96_192 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 192 \
--e_layers 2 \
--enc_in 8 \
--dec_in 8 \
--c_out 8 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/exchange_rate/ \
--data_path exchange_rate.csv \
--model_id Exchange_96_336 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 336 \
--e_layers 2 \
--enc_in 8 \
--dec_in 8 \
--c_out 8 \
--des 'Exp' \
--itr 1 \
--d_model 128 \
--d_ff 128 \
--train_epochs 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/exchange_rate/ \
--data_path exchange_rate.csv \
--model_id Exchange_96_720 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 720 \
--e_layers 2 \
--enc_in 8 \
--dec_in 8 \
--c_out 8 \
--des 'Exp' \
--d_model 128 \
--d_ff 128 \
--itr 1
\ No newline at end of file
export CUDA_VISIBLE_DEVICES=1
model_name=iTransformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS03.npz \
--model_id PEMS03_96_12 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 12 \
--e_layers 4 \
--enc_in 358 \
--dec_in 358 \
--c_out 358 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--learning_rate 0.001 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS03.npz \
--model_id PEMS03_96_24 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 24 \
--e_layers 4 \
--enc_in 358 \
--dec_in 358 \
--c_out 358 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--learning_rate 0.001 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS03.npz \
--model_id PEMS03_96_48 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 48 \
--e_layers 4 \
--enc_in 358 \
--dec_in 358 \
--c_out 358 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--learning_rate 0.001 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS03.npz \
--model_id PEMS03_96_96 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 96 \
--e_layers 4 \
--enc_in 358 \
--dec_in 358 \
--c_out 358 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--learning_rate 0.001 \
--itr 1
\ No newline at end of file
export CUDA_VISIBLE_DEVICES=2
model_name=iTransformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS04.npz \
--model_id PEMS04_96_12 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 12 \
--e_layers 4 \
--enc_in 307 \
--dec_in 307 \
--c_out 307 \
--des 'Exp' \
--d_model 1024 \
--d_ff 1024 \
--learning_rate 0.0005 \
--itr 1 \
--use_norm 0
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS04.npz \
--model_id PEMS04_96_24 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 24 \
--e_layers 4 \
--enc_in 307 \
--dec_in 307 \
--c_out 307 \
--des 'Exp' \
--d_model 1024 \
--d_ff 1024 \
--learning_rate 0.0005 \
--itr 1 \
--use_norm 0
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS04.npz \
--model_id PEMS04_96_48 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 48 \
--e_layers 4 \
--enc_in 307 \
--dec_in 307 \
--c_out 307 \
--des 'Exp' \
--d_model 1024 \
--d_ff 1024 \
--learning_rate 0.0005 \
--itr 1 \
--use_norm 0
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS04.npz \
--model_id PEMS04_96_96 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 96 \
--e_layers 4 \
--enc_in 307 \
--dec_in 307 \
--c_out 307 \
--des 'Exp' \
--d_model 1024 \
--d_ff 1024 \
--learning_rate 0.0005 \
--itr 1 \
--use_norm 0
export CUDA_VISIBLE_DEVICES=2
model_name=iTransformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS07.npz \
--model_id PEMS07_96_12 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 12 \
--e_layers 2 \
--enc_in 883 \
--dec_in 883 \
--c_out 883 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--learning_rate 0.001 \
--itr 1 \
--use_norm 0
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS07.npz \
--model_id PEMS07_96_24 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 24 \
--e_layers 2 \
--enc_in 883 \
--dec_in 883 \
--c_out 883 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--learning_rate 0.001 \
--itr 1 \
--use_norm 0
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS07.npz \
--model_id PEMS07_96_48 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 48 \
--e_layers 4 \
--enc_in 883 \
--dec_in 883 \
--c_out 883 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--batch_size 16\
--learning_rate 0.001 \
--itr 1 \
--use_norm 0
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS07.npz \
--model_id PEMS07_96_96 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 96 \
--e_layers 4 \
--enc_in 883 \
--dec_in 883 \
--c_out 883 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--batch_size 16\
--learning_rate 0.001 \
--itr 1 \
--use_norm 0
export CUDA_VISIBLE_DEVICES=2
model_name=iTransformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS08.npz \
--model_id PEMS08_96_12 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 12 \
--e_layers 2 \
--enc_in 170 \
--dec_in 170 \
--c_out 170 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--itr 1 \
--use_norm 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS08.npz \
--model_id PEMS08_96_24 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 24 \
--e_layers 2 \
--enc_in 170 \
--dec_in 170 \
--c_out 170 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--itr 1 \
--use_norm 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS08.npz \
--model_id PEMS08_96_48 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 48 \
--e_layers 4 \
--enc_in 170 \
--dec_in 170 \
--c_out 170 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--batch_size 16\
--learning_rate 0.001 \
--itr 1 \
--use_norm 0
python -u run.py \
--is_training 1 \
--root_path ./dataset/PEMS/ \
--data_path PEMS08.npz \
--model_id PEMS08_96_96 \
--model $model_name \
--data PEMS \
--features M \
--seq_len 96 \
--pred_len 96 \
--e_layers 4 \
--enc_in 170 \
--dec_in 170 \
--c_out 170 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--batch_size 16\
--learning_rate 0.001 \
--itr 1 \
--use_norm 0
# iTransformer for Multivariate Time Series Forecasting
This folder contains the reproductions of the iTransformers for Multivariate Time Series Forecasting (MTSF).
## Dataset
Extensive challenging multivariate forecasting tasks are evaluated as the benchmark. We provide the download links: [Google Drive](https://drive.google.com/file/d/1l51QsKvQPcqILT3DwfjCgx8Dsg2rpjot/view?usp=drive_link) or [Tsinghua Cloud](https://cloud.tsinghua.edu.cn/f/2ea5ca3d621e4e5ba36a/).
<p align="center">
<img src="../..//figures/datasets_mtsf.png" alt="" align=center />
</p>
## Scripts
In each folder named after the dataset, we provide the iTransformer experiments under four different prediction lengths as shown in the table above.
```
# iTransformer on the Traffic Dataset
bash ./scripts/multivariate_forecasting/Traffic/iTransformer.sh
```
To evaluate the model under other input/prediction lengths, feel free to change the ```seq_len``` and ```pred_len``` arguments:
```
# iTransformer on the Electricity Dataset, where 180 time steps are inputted as the observations, and the task is to predict the future 60 steps
python -u run.py \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_180_60 \
--model $model_name \
--data custom \
--features M \
--seq_len 180 \
--pred_len 60 \
--e_layers 3 \
--enc_in 321 \
--dec_in 321 \
--c_out 321 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--batch_size 16 \
--learning_rate 0.0005 \
--itr 1
```
## Training on Custom Dataset
To train with your own time series dataset, you can try out the following steps:
1. Read through the ```Dataset_Custom``` class under the ```data_provider/data_loader``` folder, which provides the functionality to load and process time series files.
2. The file should be ```csv``` format with the first column containing the timestamp and the following columns containing the variates of time series.
3. Set ```data=custom``` and modify the ```enc_in```, ```dec_in```, ```c_out``` arguments according to your number of variates in the training script.
export CUDA_VISIBLE_DEVICES=1
model_name=iTransformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/Solar/ \
--data_path solar_AL.txt \
--model_id solar_96_96 \
--model $model_name \
--data Solar \
--features M \
--seq_len 96 \
--pred_len 96 \
--e_layers 2 \
--enc_in 137 \
--dec_in 137 \
--c_out 137 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--learning_rate 0.0005 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/Solar/ \
--data_path solar_AL.txt \
--model_id solar_96_192 \
--model $model_name \
--data Solar \
--features M \
--seq_len 96 \
--pred_len 192 \
--e_layers 2 \
--enc_in 137 \
--dec_in 137 \
--c_out 137 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--learning_rate 0.0005 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/Solar/ \
--data_path solar_AL.txt \
--model_id solar_96_336 \
--model $model_name \
--data Solar \
--features M \
--seq_len 96 \
--pred_len 336 \
--e_layers 2 \
--enc_in 137 \
--dec_in 137 \
--c_out 137 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--learning_rate 0.0005 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/Solar/ \
--data_path solar_AL.txt \
--model_id solar_96_720 \
--model $model_name \
--data Solar \
--features M \
--seq_len 96 \
--pred_len 720 \
--e_layers 2 \
--enc_in 137 \
--dec_in 137 \
--c_out 137 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--learning_rate 0.0005 \
--itr 1
export CUDA_VISIBLE_DEVICES=2
model_name=iTransformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/traffic/ \
--data_path traffic.csv \
--model_id traffic_96_96 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 96 \
--e_layers 4 \
--enc_in 862 \
--dec_in 862 \
--c_out 862 \
--des 'Exp' \
--d_model 512\
--d_ff 512 \
--batch_size 16 \
--learning_rate 0.001 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/traffic/ \
--data_path traffic.csv \
--model_id traffic_96_192 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 192 \
--e_layers 4 \
--enc_in 862 \
--dec_in 862 \
--c_out 862 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--batch_size 16 \
--learning_rate 0.001 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/traffic/ \
--data_path traffic.csv \
--model_id traffic_96_336 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 336 \
--e_layers 4 \
--enc_in 862 \
--dec_in 862 \
--c_out 862 \
--des 'Exp' \
--d_model 512\
--d_ff 512 \
--batch_size 16 \
--learning_rate 0.001 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/traffic/ \
--data_path traffic.csv \
--model_id traffic_96_720 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 720 \
--e_layers 4 \
--enc_in 862 \
--dec_in 862 \
--c_out 862 \
--des 'Exp' \
--d_model 512 \
--d_ff 512 \
--batch_size 16 \
--learning_rate 0.001\
--itr 1
\ No newline at end of file
export CUDA_VISIBLE_DEVICES=1
model_name=iTransformer
python -u run.py \
--is_training 1 \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather_96_96 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 96 \
--e_layers 3 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--d_model 512\
--d_ff 512\
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather_96_192 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 192 \
--e_layers 3 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--d_model 512\
--d_ff 512\
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather_96_336 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 336 \
--e_layers 3 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--d_model 512\
--d_ff 512\
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather_96_720 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--pred_len 720 \
--e_layers 3 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--d_model 512\
--d_ff 512\
--itr 1
\ No newline at end of file
export CUDA_VISIBLE_DEVICES=0
model_name=Flowformer
#python -u run.py \
## --is_training 1 \
# --root_path ./dataset/electricity/ \
# --data_path electricity.csv \
# --model_id ECL_96_96 \
# --model $model_name \
# --data custom \
# --features M \
# --seq_len 96 \
# --label_len 48 \
# --pred_len 96 \
# --e_layers 2 \
# --d_layers 1 \
# --factor 3 \
# --enc_in 321 \
# --dec_in 321 \
# --c_out 321 \
# --des 'Exp' \
# --itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_96_96 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 48 \
--pred_len 96 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 64 \
--dec_in 64 \
--c_out 64 \
--des 'Exp' \
--channel_independence true \
--exp_name partial_train \
--batch_size 8 \
--d_model 32 \
--d_ff 64 \
--itr 1
model_name=iFlowformer
#python -u run.py \
## --is_training 1 \
# --root_path ./dataset/electricity/ \
# --data_path electricity.csv \
# --model_id ECL_96_96 \
# --model $model_name \
# --data custom \
# --features M \
# --seq_len 96 \
# --label_len 48 \
# --pred_len 96 \
# --e_layers 2 \
# --d_layers 1 \
# --factor 3 \
# --enc_in 321 \
# --dec_in 321 \
# --c_out 321 \
# --des 'Exp' \
# --itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_96_96 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 48 \
--pred_len 96 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 64 \
--dec_in 64 \
--c_out 64 \
--des 'Exp' \
--exp_name partial_train \
--itr 1
export CUDA_VISIBLE_DEVICES=1
model_name=Informer
#python -u run.py \
## --is_training 1 \
# --root_path ./dataset/electricity/ \
# --data_path electricity.csv \
# --model_id ECL_96_96 \
# --model $model_name \
# --data custom \
# --features M \
# --seq_len 96 \
# --label_len 48 \
# --pred_len 96 \
# --e_layers 2 \
# --d_layers 1 \
# --factor 3 \
# --enc_in 321 \
# --dec_in 321 \
# --c_out 321 \
# --des 'Exp' \
# --itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_96_96 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 48 \
--pred_len 96 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 64 \
--dec_in 64 \
--c_out 64 \
--des 'Exp' \
--channel_independence true \
--exp_name partial_train \
--batch_size 8 \
--d_model 32 \
--d_ff 64 \
--itr 1
model_name=iInformer
#python -u run.py \
## --is_training 1 \
# --root_path ./dataset/electricity/ \
# --data_path electricity.csv \
# --model_id ECL_96_96 \
# --model $model_name \
# --data custom \
# --features M \
# --seq_len 96 \
# --label_len 48 \
# --pred_len 96 \
# --e_layers 2 \
# --d_layers 1 \
# --factor 3 \
# --enc_in 321 \
# --dec_in 321 \
# --c_out 321 \
# --des 'Exp' \
# --itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_96_96 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 48 \
--pred_len 96 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 64 \
--dec_in 64 \
--c_out 64 \
--des 'Exp' \
--exp_name partial_train \
--itr 1
\ No newline at end of file
export CUDA_VISIBLE_DEVICES=2
model_name=Reformer
#python -u run.py \
## --is_training 1 \
# --root_path ./dataset/electricity/ \
# --data_path electricity.csv \
# --model_id ECL_96_96 \
# --model $model_name \
# --data custom \
# --features M \
# --seq_len 96 \
# --label_len 48 \
# --pred_len 96 \
# --e_layers 2 \
# --d_layers 1 \
# --factor 3 \
# --enc_in 321 \
# --dec_in 321 \
# --c_out 321 \
# --des 'Exp' \
# --itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_96_96 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 48 \
--pred_len 96 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 64 \
--dec_in 64 \
--c_out 64 \
--des 'Exp' \
--channel_independence true \
--exp_name partial_train \
--batch_size 8 \
--d_model 32 \
--d_ff 64 \
--itr 1
model_name=iReformer
#python -u run.py \
## --is_training 1 \
# --root_path ./dataset/electricity/ \
# --data_path electricity.csv \
# --model_id ECL_96_96 \
# --model $model_name \
# --data custom \
# --features M \
# --seq_len 96 \
# --label_len 48 \
# --pred_len 96 \
# --e_layers 2 \
# --d_layers 1 \
# --factor 3 \
# --enc_in 321 \
# --dec_in 321 \
# --c_out 321 \
# --des 'Exp' \
# --itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_96_96 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 48 \
--pred_len 96 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 64 \
--dec_in 64 \
--c_out 64 \
--des 'Exp' \
--exp_name partial_train \
--itr 1
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment