wassname/ETSformer
1
0
Fork 0
mirror of https://github.com/wassname/ETSformer.git synced 2026-08-20 12:00:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

first commit

Browse Source
This commit is contained in:
gorold
2022-06-20 15:15:36 +08:00
commit dcc5a96275
36 changed files with 2682 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/ECL.sh
+77
View File
@@ -0,0 +1,77 @@
python -u run.py \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL \
--model ETSformer \
--data custom \
--features M \
--seq_len 336 \
--pred_len 96 \
--e_layers 2 \
--d_layers 2 \
--enc_in 321 \
--dec_in 321 \
--c_out 321 \
--des 'Exp' \
--K 3 \
--learning_rate 3e-4 \
--itr 1
python -u run.py \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL \
--model ETSformer \
--data custom \
--features M \
--seq_len 336 \
--pred_len 192 \
--e_layers 2 \
--d_layers 2 \
--enc_in 321 \
--dec_in 321 \
--c_out 321 \
--des 'Exp' \
--K 3 \
--learning_rate 3e-4 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL \
--model ETSformer \
--data custom \
--features M \
--seq_len 336 \
--pred_len 336 \
--e_layers 2 \
--d_layers 2 \
--enc_in 321 \
--dec_in 321 \
--c_out 321 \
--des 'Exp' \
--K 3 \
--learning_rate 3e-4 \
--itr 1
python -u run.py \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL \
--model ETSformer \
--data custom \
--features M \
--seq_len 336 \
--pred_len 720 \
--e_layers 2 \
--d_layers 2 \
--enc_in 321 \
--dec_in 321 \
--c_out 321 \
--des 'Exp' \
--K 3 \
--learning_rate 1e-4 \
--itr 1
scripts/ETTm2.sh
+75
View File
@@ -0,0 +1,75 @@
python -u run.py \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2 \
--model ETSformer \
--data ETTm2 \
--features M \
--seq_len 96 \
--pred_len 96 \
--e_layers 2 \
--d_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--K 3 \
--learning_rate 1e-5 \
--itr 1
python -u run.py \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2 \
--model ETSformer \
--data ETTm2 \
--features M \
--seq_len 96 \
--pred_len 192 \
--e_layers 2 \
--d_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--K 3 \
--learning_rate 1e-5 \
--itr 1
python -u run.py \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2 \
--model ETSformer \
--data ETTm2 \
--features M \
--seq_len 96 \
--pred_len 336 \
--e_layers 2 \
--d_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--K 3 \
--learning_rate 1e-5 \
--itr 1
python -u run.py \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2 \
--model ETSformer \
--data ETTm2 \
--features M \
--seq_len 96 \
--pred_len 720 \
--e_layers 2 \
--d_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--K 3 \
--learning_rate 1e-5 \
--itr 1
scripts/ETTm2_univar.sh
+75
View File
@@ -0,0 +1,75 @@
python -u run.py \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2 \
--model ETSformer \
--data ETTm2 \
--features S \
--seq_len 96 \
--pred_len $96 \
--e_layers 2 \
--d_layers 2 \
--enc_in 1 \
--dec_in 1 \
--c_out 1 \
--des 'Exp' \
--K 2 \
--learning_rate 1e-5 \
--itr 1
python -u run.py \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2 \
--model ETSformer \
--data ETTm2 \
--features S \
--seq_len 96 \
--pred_len 192 \
--e_layers 2 \
--d_layers 2 \
--enc_in 1 \
--dec_in 1 \
--c_out 1 \
--des 'Exp' \
--K 2 \
--learning_rate 1e-3 \
--itr 1
python -u run.py \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2 \
--model ETSformer \
--data ETTm2 \
--features S \
--seq_len 96 \
--pred_len 336 \
--e_layers 2 \
--d_layers 2 \
--enc_in 1 \
--dec_in 1 \
--c_out 1 \
--des 'Exp' \
--K 1 \
--learning_rate 1e-3 \
--itr 1
python -u run.py \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2 \
--model ETSformer \
--data ETTm2 \
--features S \
--seq_len 720 \
--pred_len 720 \
--e_layers 2 \
--d_layers 2 \
--enc_in 1 \
--dec_in 1 \
--c_out 1 \
--des 'Exp' \
--K 3 \
--learning_rate 3e-4 \
--itr 1
scripts/Exchange.sh
+75
View File
@@ -0,0 +1,75 @@
python -u run.py \
--root_path ./dataset/exchange_rate/ \
--data_path exchange_rate.csv \
--model_id Exchange \
--model ETSformer \
--data custom \
--features M \
--seq_len 336 \
--pred_len 96 \
--e_layers 2 \
--d_layers 2 \
--enc_in 8 \
--dec_in 8 \
--c_out 8 \
--des 'Exp' \
--K 0 \
--learning_rate 3e-5 \
--itr 1
python -u run.py \
--root_path ./dataset/exchange_rate/ \
--data_path exchange_rate.csv \
--model_id Exchange \
--model ETSformer \
--data custom \
--features M \
--seq_len 720 \
--pred_len 192 \
--e_layers 2 \
--d_layers 2 \
--enc_in 8 \
--dec_in 8 \
--c_out 8 \
--des 'Exp' \
--K 0 \
--learning_rate 1e-4 \
--itr 1
python -u run.py \
--root_path ./dataset/exchange_rate/ \
--data_path exchange_rate.csv \
--model_id Exchange \
--model ETSformer \
--data custom \
--features M \
--seq_len 336 \
--pred_len 336 \
--e_layers 2 \
--d_layers 2 \
--enc_in 8 \
--dec_in 8 \
--c_out 8 \
--des 'Exp' \
--K 0 \
--learning_rate 3e-4 \
--itr 1
python -u run.py \
--root_path ./dataset/exchange_rate/ \
--data_path exchange_rate.csv \
--model_id Exchange \
--model ETSformer \
--data custom \
--features M \
--seq_len 192 \
--pred_len 720 \
--e_layers 2 \
--d_layers 2 \
--enc_in 8 \
--dec_in 8 \
--c_out 8 \
--des 'Exp' \
--K 0 \
--learning_rate 1e-3 \
--itr 1
scripts/Exchange_univar.sh
+76
View File
@@ -0,0 +1,76 @@
python -u run.py \
--root_path ./dataset/exchange_rate/ \
--data_path exchange_rate.csv \
--model_id Exchange \
--model ETSformer \
--data custom \
--features S \
--seq_len 96 \
--pred_len 96 \
--e_layers 2 \
--d_layers 2 \
--enc_in 1 \
--dec_in 1 \
--c_out 1 \
--des 'Exp' \
--K 0 \
--learning_rate 1e-3 \
--itr 1
python -u run.py \
--root_path ./dataset/exchange_rate/ \
--data_path exchange_rate.csv \
--model_id Exchange \
--model ETSformer \
--data custom \
--features S \
--seq_len 96 \
--pred_len 192 \
--e_layers 2 \
--d_layers 2 \
--enc_in 1 \
--dec_in 1 \
--c_out 1 \
--des 'Exp' \
--K 0 \
--learning_rate 1e-3 \
--itr 1
python -u run.py \
--root_path ./dataset/exchange_rate/ \
--data_path exchange_rate.csv \
--model_id Exchange \
--model ETSformer \
--data custom \
--features S \
--seq_len 192 \
--pred_len 336 \
--e_layers 2 \
--d_layers 2 \
--enc_in 1 \
--dec_in 1 \
--c_out 1 \
--des 'Exp' \
--K 1 \
--learning_rate 3e-4 \
--itr 1
python -u run.py \
--root_path ./dataset/exchange_rate/ \
--data_path exchange_rate.csv \
--model_id Exchange \
--model ETSformer \
--data custom \
--features S \
--seq_len 720 \
--pred_len 720 \
--e_layers 2 \
--d_layers 2 \
--enc_in 1 \
--dec_in 1 \
--c_out 1 \
--des 'Exp' \
--K 1 \
--learning_rate 3e-5 \
--itr 1
scripts/ILI.sh
+75
View File
@@ -0,0 +1,75 @@
python -u run.py \
--root_path ./dataset/illness/ \
--data_path national_illness.csv \
--model_id ili \
--model ETSformer \
--data custom \
--features M \
--seq_len 48 \
--pred_len 24 \
--e_layers 2 \
--d_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--K 1 \
--learning_rate 1e-3 \
--itr 1
python -u run.py \
--root_path ./dataset/illness/ \
--data_path national_illness.csv \
--model_id ili \
--model ETSformer \
--data custom \
--features M \
--seq_len 60 \
--pred_len 36 \
--e_layers 2 \
--d_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--K 1 \
--learning_rate 1e-3 \
--itr 1
python -u run.py \
--root_path ./dataset/illness/ \
--data_path national_illness.csv \
--model_id ili \
--model ETSformer \
--data custom \
--features M \
--seq_len 60 \
--pred_len 48 \
--e_layers 2 \
--d_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--K 1 \
--learning_rate 1e-3 \
--itr 1
python -u run.py \
--root_path ./dataset/illness/ \
--data_path national_illness.csv \
--model_id ili \
--model ETSformer \
--data custom \
--features M \
--seq_len 60 \
--pred_len 60 \
--e_layers 2 \
--d_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--K 1 \
--learning_rate 1e-3 \
--itr 1
scripts/Traffic.sh
+75
View File
@@ -0,0 +1,75 @@
python -u run.py \
--root_path ./dataset/traffic/ \
--data_path traffic.csv \
--model_id traffic \
--model ETSformer \
--data custom \
--features M \
--seq_len 336 \
--pred_len 96 \
--e_layers 2 \
--d_layers 2 \
--enc_in 862 \
--dec_in 862 \
--c_out 862 \
--des 'Exp' \
--K 3 \
--learning_rate 1e-3 \
--itr 1
python -u run.py \
--root_path ./dataset/traffic/ \
--data_path traffic.csv \
--model_id traffic \
--model ETSformer \
--data custom \
--features M \
--seq_len 336 \
--pred_len 192 \
--e_layers 2 \
--d_layers 2 \
--enc_in 862 \
--dec_in 862 \
--c_out 862 \
--des 'Exp' \
--K 3 \
--learning_rate 1e-3 \
--itr 1
python -u run.py \
--root_path ./dataset/traffic/ \
--data_path traffic.csv \
--model_id traffic \
--model ETSformer \
--data custom \
--features M \
--seq_len 336 \
--pred_len 336 \
--e_layers 2 \
--d_layers 2 \
--enc_in 862 \
--dec_in 862 \
--c_out 862 \
--des 'Exp' \
--K 3 \
--learning_rate 1e-3 \
--itr 1
python -u run.py \
--root_path ./dataset/traffic/ \
--data_path traffic.csv \
--model_id traffic \
--model ETSformer \
--data custom \
--features M \
--seq_len 336 \
--pred_len 720 \
--e_layers 2 \
--d_layers 2 \
--enc_in 862 \
--dec_in 862 \
--c_out 862 \
--des 'Exp' \
--K 3 \
--learning_rate 1e-3 \
--itr 1
scripts/Weather.sh
+75
View File
@@ -0,0 +1,75 @@
python -u run.py \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather \
--model ETSformer \
--data custom \
--features M \
--seq_len 192 \
--pred_len 96 \
--e_layers 2 \
--d_layers 2 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--K 1 \
--learning_rate 1e-3 \
--itr 1
python -u run.py \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather \
--model ETSformer \
--data custom \
--features M \
--seq_len 192 \
--pred_len 192 \
--e_layers 2 \
--d_layers 2 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--K 1 \
--learning_rate 1e-3 \
--itr 1
python -u run.py \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather \
--model ETSformer \
--data custom \
--features M \
--seq_len 720 \
--pred_len 336 \
--e_layers 2 \
--d_layers 2 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--K 3 \
--learning_rate 3e-4 \
--itr 1
python -u run.py \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather \
--model ETSformer \
--data custom \
--features M \
--seq_len 720 \
--pred_len 720 \
--e_layers 2 \
--d_layers 2 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--K 3 \
--learning_rate 3e-4 \
--itr 1
scripts/grid_search.sh
+171
View File
@@ -0,0 +1,171 @@
for lr in 1e-5 3e-5 1e-4 3e-4 1e-3; do
for k in 0 1 2 3; do
for pl in 96 192 336 720; do
for sl in 96 192 336 720; do
# ETTm2
python -u run.py \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2 \
--model ETSformer \
--data ETTm2 \
--features M \
--seq_len ${sl} \
--pred_len ${pl} \
--e_layers 2 \
--d_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--K ${k} \
--learning_rate ${lr} \
--itr 3
# ETTm2 univar
python -u run.py \
--root_path ./dataset/ETT-small/ \
--data_path ETTm2.csv \
--model_id ETTm2 \
--model ETSformer \
--data ETTm2 \
--features S \
--seq_len ${sl} \
--pred_len ${pl} \
--e_layers 2 \
--d_layers 2 \
--enc_in 1 \
--dec_in 1 \
--c_out 1 \
--des 'Exp' \
--K ${k} \
--learning_rate ${lr} \
--itr 3
# ECL
python -u run.py \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL \
--model ETSformer \
--data custom \
--features M \
--seq_len ${sl} \
--pred_len ${pl} \
--e_layers 2 \
--d_layers 2 \
--enc_in 321 \
--dec_in 321 \
--c_out 321 \
--des 'Exp' \
--K ${k} \
--learning_rate ${lr} \
--itr 3
# Traffic
python -u run.py \
--root_path ./dataset/traffic/ \
--data_path traffic.csv \
--model_id traffic \
--model ETSformer \
--data custom \
--features M \
--seq_len ${sl} \
--pred_len ${pl} \
--e_layers 2 \
--d_layers 2 \
--enc_in 862 \
--dec_in 862 \
--c_out 862 \
--des 'Exp' \
--K ${k} \
--learning_rate ${lr} \
--itr 3
# Weather
python -u run.py \
--root_path ./dataset/weather/ \
--data_path weather.csv \
--model_id weather \
--model ETSformer \
--data custom \
--features M \
--seq_len ${sl} \
--pred_len ${pl} \
--e_layers 2 \
--d_layers 2 \
--enc_in 21 \
--dec_in 21 \
--c_out 21 \
--des 'Exp' \
--K ${k} \
--learning_rate ${lr} \
--itr 3
# Exchange
python -u run.py \
--root_path ./dataset/exchange_rate/ \
--data_path exchange_rate.csv \
--model_id Exchange \
--model ETSformer \
--data custom \
--features M \
--seq_len ${sl} \
--pred_len ${pl} \
--e_layers 2 \
--d_layers 2 \
--enc_in 8 \
--dec_in 8 \
--c_out 8 \
--des 'Exp' \
--K ${k} \
--learning_rate ${lr} \
--itr 3
# Exchange univar
python -u run.py \
--root_path ./dataset/exchange_rate/ \
--data_path exchange_rate.csv \
--model_id Exchange \
--model ETSformer \
--data custom \
--features S \
--seq_len ${sl} \
--pred_len ${pl} \
--e_layers 2 \
--d_layers 2 \
--enc_in 1 \
--dec_in 1 \
--c_out 1 \
--des 'Exp' \
--K ${k} \
--learning_rate ${lr} \
--itr 3
done
done
# ILI
for pl in 24 36 48 60; do
for sl in 24 36 48 60; do
python -u run.py \
--root_path ./dataset/illness/ \
--data_path national_illness.csv \
--model_id ili \
--model ETSformer \
--data custom \
--features M \
--seq_len ${sl} \
--pred_len ${pl} \
--e_layers 2 \
--d_layers 2 \
--enc_in 7 \
--dec_in 7 \
--c_out 7 \
--des 'Exp' \
--K ${k} \
--learning_rate ${lr} \
--itr 3
done
done
done
done