SpikF-GO / scripts /ecl.sh
bakhshaliyev's picture
Upload folder using huggingface_hub
d88ab6d verified
Raw
History Blame Contribute Delete
4.73 kB
#!/bin/bash
if [ ! -d "./logs" ]; then
mkdir ./logs
fi
if [ ! -d "./logs/LongForecasting" ]; then
mkdir ./logs/LongForecasting
fi
python train.py \
--model FGN \
--data electricity \
--feature_size 370\
--embed_size 128 \
--hidden_size 256 \
--batch_size 16 \
--train_ratio 0.7 \
--val_ratio 0.2 \
--seq_length 12 \
--pre_length 12 \
--train_epochs 100 \
--learning_rate 0.00001 \
--device cuda:0 >logs/LongForecasting/ECL_FGN.log
python train.py \
--model SpikF \
--data electricity \
--feature_size 370\
--embed_size 128 \
--hidden_size 256 \
--batch_size 16 \
--train_ratio 0.7 \
--val_ratio 0.2 \
--seq_length 12 \
--pre_length 12 \
--train_epochs 100 \
--learning_rate 0.00001 \
--T 16 \
--blocks 2\
--device cuda:0 >logs/LongForecasting/ECL_SpikF.log
python train.py \
--model iSpikformer \
--data electricity \
--feature_size 370\
--embed_size 128 \
--hidden_size 256 \
--batch_size 16 \
--train_ratio 0.7 \
--val_ratio 0.2 \
--seq_length 12 \
--pre_length 12 \
--train_epochs 100 \
--learning_rate 0.00001 \
--blocks 2 \
--device cuda:0 >logs/LongForecasting/ECL_iSpikformer.log
python train.py \
--model SpikF_GO \
--data electricity \
--feature_size 370\
--embed_size 128 \
--hidden_size 256 \
--batch_size 16 \
--train_ratio 0.7 \
--val_ratio 0.2 \
--seq_length 12 \
--pre_length 12 \
--train_epochs 100 \
--learning_rate 0.00001 \
--energy_loss True \
--device cuda:0 >logs/LongForecasting/ECL_SpikFGO.log
python train.py \
--model SpikF_GO_CPG \
--data electricity \
--feature_size 370\
--embed_size 128 \
--hidden_size 256 \
--batch_size 16 \
--train_ratio 0.7 \
--val_ratio 0.2 \
--seq_length 12 \
--pre_length 12 \
--train_epochs 100 \
--learning_rate 0.00001 \
--energy_loss True \
--device cuda:0 >logs/LongForecasting/ECL_SpikFGOCPG.log
python train.py \
--model SpikeRNN_CPG \
--data electricity \
--feature_size 370\
--embed_size 128 \
--hidden_size 128\
--batch_size 16 \
--train_ratio 0.7 \
--val_ratio 0.2 \
--seq_length 12 \
--pre_length 12 \
--train_epochs 100 \
--learning_rate 0.00001 \
--blocks 2 \
--device cuda:0 >logs/LongForecasting/ECL_SpikeRNNCPG.log
python train.py \
--model SpikeGRU \
--data electricity \
--feature_size 370\
--embed_size 128 \
--hidden_size 64 \
--batch_size 16 \
--train_ratio 0.7 \
--val_ratio 0.2 \
--seq_length 12 \
--pre_length 12 \
--train_epochs 100 \
--learning_rate 0.00001 \
--device cuda:0 >logs/LongForecasting/ECL_SpikeGRU.log
python train.py \
--model SpikeTCN_CPG \
--data electricity \
--feature_size 370\
--embed_size 128 \
--hidden_size 64\
--batch_size 16 \
--train_ratio 0.7 \
--val_ratio 0.2 \
--seq_length 12 \
--pre_length 12 \
--train_epochs 100 \
--learning_rate 0.00001 \
--blocks 3\
--device cuda:0 >logs/LongForecasting/ECL_SpikeTCNCPG.log
python train.py \
--model Spikformer_CPG \
--data electricity \
--feature_size 370\
--embed_size 128 \
--hidden_size 128\
--batch_size 16 \
--train_ratio 0.7 \
--val_ratio 0.2 \
--seq_length 12 \
--pre_length 12 \
--train_epochs 100 \
--learning_rate 0.00001 \
--blocks 2 \
--device cuda:0 >logs/LongForecasting/ECL_SpikformerCPG.log
python train.py \
--model TSTCN \
--data electricity \
--feature_size 370\
--embed_size 128 \
--hidden_size 64 \
--batch_size 16 \
--train_ratio 0.7 \
--val_ratio 0.2 \
--seq_length 12 \
--pre_length 12 \
--train_epochs 100 \
--learning_rate 0.00001 \
--kernel_size 3\
--blocks 3 \
--device cuda:0 >logs/LongForecasting/ECL_TSTCN.log
python train.py \
--model TSGRU \
--data electricity \
--feature_size 370\
--embed_size 128 \
--hidden_size 64 \
--batch_size 16 \
--train_ratio 0.7 \
--val_ratio 0.2 \
--seq_length 12 \
--pre_length 12 \
--train_epochs 100 \
--learning_rate 0.00001 \
--device cuda:0 >logs/LongForecasting/ECL_TSGRU.log
python train.py \
--model TSFormer \
--data electricity \
--feature_size 370\
--embed_size 128 \
--hidden_size 64 \
--batch_size 16 \
--train_ratio 0.7 \
--val_ratio 0.2 \
--seq_length 12 \
--pre_length 12 \
--train_epochs 100 \
--learning_rate 0.00001 \
--device cuda:0 >logs/LongForecasting/ECL_TSFormer.log