File size: 475 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 |
export CUDA_VISIBLE_DEVICES=6
model_name=DLinear
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 \
--loss 'SMAPE' |