File size: 573 Bytes
f9d3aeb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | export CUDA_VISIBLE_DEVICES=6
# model_name=
for model_name in DLinear TimesNet ETSformer Autoformer Informer Reformer
do
python -u run.py \
--task_name short_term_forecast \
--is_training 1 \
--root_path ./dataset/poly \
--seasonal_patterns 'Daily' \
--model_id poly_Daily \
--model $model_name \
--data poly \
--features M \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 1 \
--dec_in 1 \
--c_out 1 \
--batch_size 128 \
--d_model 128 \
--des 'Exp' \
--itr 1 \
--learning_rate 0.001
done |