first commit

scripts/long_term_forecast/ETT_script/SegRNN_ETTm1.sh

@@ -0,0 +1,26 @@
+export CUDA_VISIBLE_DEVICES=2
+
+model_name=SegRNN
+
+seq_len=96
+for pred_len in 96 192 336 720
+do
+python -u run.py \
+  --task_name long_term_forecast \
+  --is_training 1 \
+  --root_path ./dataset/ETT-small/ \
+  --data_path ETTm1.csv \
+  --model_id ETTm1_$seq_len'_'$pred_len \
+  --model $model_name \
+  --data ETTm1 \
+  --features M \
+  --seq_len $seq_len \
+  --pred_len $pred_len \
+  --seg_len 48 \
+  --enc_in 7 \
+  --d_model 512 \
+  --dropout 0.5 \
+  --learning_rate 0.0001 \
+  --des 'Exp' \
+  --itr 1
+done