TempoPFN / config.json
vlad-moroshan's picture
Create config.json
2ce3135 verified
Raw History Blame Contribute Delete
934 Bytes
{
"model_type": "tempopfn",
"architectures": ["TempoPFN"],
"TimeSeriesModel": {
"embed_size": 512,
"num_encoder_layers": 10,
"scaler": "custom_robust",
"epsilon": 0.00001,
"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": 4,
"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]
}
}