File size: 2,113 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 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 |
export CUDA_VISIBLE_DEVICES=7
model_name=TimesNet
python -u run.py \
--task_name imputation \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_mask_0.125 \
--mask_rate 0.125 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 0 \
--pred_len 0 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 321 \
--dec_in 321 \
--c_out 321 \
--batch_size 16 \
--d_model 64 \
--d_ff 64 \
--des 'Exp' \
--itr 1 \
--top_k 3 \
--learning_rate 0.001
python -u run.py \
--task_name imputation \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_mask_0.25 \
--mask_rate 0.25 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 0 \
--pred_len 0 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 321 \
--dec_in 321 \
--c_out 321 \
--batch_size 16 \
--d_model 64 \
--d_ff 64 \
--des 'Exp' \
--itr 1 \
--top_k 3 \
--learning_rate 0.001
python -u run.py \
--task_name imputation \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_mask_0.375 \
--mask_rate 0.375 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 0 \
--pred_len 0 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 321 \
--dec_in 321 \
--c_out 321 \
--batch_size 16 \
--d_model 64 \
--d_ff 64 \
--des 'Exp' \
--itr 1 \
--top_k 3 \
--learning_rate 0.001
python -u run.py \
--task_name imputation \
--is_training 1 \
--root_path ./dataset/electricity/ \
--data_path electricity.csv \
--model_id ECL_mask_0.5 \
--mask_rate 0.5 \
--model $model_name \
--data custom \
--features M \
--seq_len 96 \
--label_len 0 \
--pred_len 0 \
--e_layers 2 \
--d_layers 1 \
--factor 3 \
--enc_in 321 \
--dec_in 321 \
--c_out 321 \
--batch_size 16 \
--d_model 64 \
--d_ff 64 \
--des 'Exp' \
--itr 1 \
--top_k 3 \
--learning_rate 0.001
|