vlad-moroshan commited on
Commit
2ce3135
·
verified ·
1 Parent(s): 24e5642

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +36 -0
config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "tempopfn",
3
+ "architectures": ["TempoPFN"],
4
+ "TimeSeriesModel": {
5
+ "embed_size": 512,
6
+ "num_encoder_layers": 10,
7
+ "scaler": "custom_robust",
8
+ "epsilon": 0.00001,
9
+ "scaler_clamp_value": null,
10
+ "handle_constants": false,
11
+ "K_max": 25,
12
+ "time_feature_config": {
13
+ "use_enhanced_features": true,
14
+ "use_holiday_features": false,
15
+ "use_index_features": true,
16
+ "include_seasonality_info": true
17
+ },
18
+ "drop_enc_allow": false,
19
+ "encoding_dropout": 0.0,
20
+ "encoder_config": {
21
+ "attn_mode": "chunk",
22
+ "num_heads": 4,
23
+ "expand_v": 1.0,
24
+ "use_short_conv": true,
25
+ "conv_size": 32,
26
+ "allow_neg_eigval": true,
27
+ "hidden_ratio": 1.0,
28
+ "use_gate": true,
29
+ "use_forget_gate": true,
30
+ "num_householder": 4,
31
+ "weaving": true
32
+ },
33
+ "loss_type": "quantile",
34
+ "quantiles": [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9]
35
+ }
36
+ }