File size: 1,500 Bytes
6dd3ebe 4cc7625 a9073bb 6dd3ebe |
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 |
autoformer_kwargs = lambda lookback,lookahead:{
'enc_in': 6,
'dec_in': 2,
'c_out': 1,
'pred_len': lookahead,
'seq_len': lookback,
'd_model': 32*4,
'data_idx': [0,3,4,5,6,7],
'time_idx': [1,2]
}
informer_kwargs = lambda lookback,lookahead:{
'enc_in': 6,
'dec_in': 2,
'c_out': 1,
'pred_len': lookahead,
'd_model': 32*4,
'data_idx': [0,3,4,5,6,7],
'time_idx': [1,2]
}
timesnet_kwargs = lambda lookback,lookahead:{
'enc_in': 6,
'dec_in': 2,
'c_out': 1,
'pred_len': lookahead,
'seq_len': lookback,
'd_model': 32*4,
'data_idx': [0,3,4,5,6,7],
'time_idx': [1,2]
}
transformer_kwargs = lambda lookback,lookahead:{
'enc_in': 6,
'dec_in': 2,
'c_out': 1,
'pred_len': lookahead,
'd_model': 32*4,
'data_idx': [0,3,4,5,6,7],
'time_idx': [1,2]
}
lstm_kwargs = lambda lookback,lookahead:{
'input_size': 8,
'hidden_size': 8*4,
'num_layers': 2,
'lookback': lookback
}
lstnet_kwargs = lambda lookback,lookahead:{
'num_features':8,
'conv1_out_channels':8*4,
'conv1_kernel_height':3*4,
'recc1_out_channels':32*4
}
patchtst_kwargs = lambda lookback,lookahead:{
'enc_in': 6,
'dec_in': 2,
'c_out': 1,
'pred_len': lookahead,
'seq_len': lookback,
'd_model': 32*4,
'data_idx': [0,3,4,5,6,7],
'time_idx': [1,2]
}
timesfm_kwargs = lambda lookback, lookahead:{
'lookback': lookback,
'lookahead': lookahead,
'context_len': 512
} |