-
Notifications
You must be signed in to change notification settings - Fork 1
/
models.sh
10 lines (9 loc) · 1.09 KB
/
models.sh
1
2
3
4
5
6
7
8
9
10
#! /bin/bash
# Autoformer
main.py --is_training 1 --root_path ./dataset/electricity/ --data_path electricity.csv --model_id ECL_720 --model Autoformer --batch_size 12 --train_epochs 3 --features SA --seq_len 720 --pred_len 720 --e_layers 2 --d_layers 1 --d_model 512 --d_ff 2048 --des 'Exp'
# Informer
main.py --is_training 1 --root_path ./dataset/electricity/ --data_path electricity.csv --model_id ECL_720 --model Informer --batch_size 12 --train_epochs 3 --features SA --seq_len 720 --pred_len 720 --e_layers 2 --d_layers 1 --d_model 512 --d_ff 2048 --des 'Exp'
# Reformer
main.py --is_training 1 --root_path ./dataset/electricity/ --data_path electricity.csv --model_id ECL_720 --model Reformer --batch_size 12 --train_epochs 3 --features SA --seq_len 720 --pred_len 720 --e_layers 2 --d_layers 1 --d_model 512 --d_ff 2048 --des 'Exp'
# Transformer
main.py --is_training 1 --root_path ./dataset/electricity/ --data_path electricity.csv --model_id ECL_720 --model Transformer --batch_size 12 --train_epochs 3 --features SA --seq_len 720 --pred_len 720 --e_layers 2 --d_layers 1 --d_model 512 --d_ff 2048 --des 'Exp'