MacroCast / config.yaml
shubhranshu's picture
Slim config.yaml to inference-only TimeSeriesModel architecture
741b113 verified
Raw
History Blame Contribute Delete
725 Bytes
TimeSeriesModel:
embed_size: 128
num_encoder_layers: 3
scaler: custom_robust
epsilon: 1.0e-05
scaler_clamp_value: null
handle_constants: false
K_max: 25
time_feature_config:
use_enhanced_features: true
use_holiday_features: false
use_index_features: true
include_seasonality_info: true
drop_enc_allow: false
encoding_dropout: 0.0
encoder_config:
attn_mode: chunk
num_heads: 2
expand_v: 1.0
use_short_conv: true
conv_size: 32
allow_neg_eigval: true
hidden_ratio: 1.0
use_gate: true
use_forget_gate: true
num_householder: 4
weaving: true
loss_type: quantile
quantiles:
- 0.1
- 0.2
- 0.3
- 0.4
- 0.5
- 0.6
- 0.7
- 0.8
- 0.9