Shoriful025 commited on
Commit
7271df1
·
verified ·
1 Parent(s): 52125e4

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +23 -0
config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "EcomSalesTrendPredictor",
3
+ "architectures": ["TimeSeriesTransformerModel"],
4
+ "model_type": "time_series_transformer",
5
+ "prediction_length": 7,
6
+ "context_length": 30,
7
+ "scaling": "std",
8
+ "input_size": 1,
9
+ "num_layers": 3,
10
+ "hidden_size": 64,
11
+ "num_heads": 4,
12
+ "output_type": "quantiles",
13
+ "lags_sequence": [1, 2, 7, 30],
14
+ "num_static_real_features": 3,
15
+ "num_dynamic_real_features": 4,
16
+ "num_static_categorical_features": 4,
17
+ "num_dynamic_categorical_features": 3,
18
+ "feat_static_cat_cardinalities": [4, 4, 2, 2],
19
+ "feat_dynamic_cat_cardinalities": [4, 4, 2],
20
+ "dropout": 0.1,
21
+ "quantiles": [0.1, 0.5, 0.9],
22
+ "torch_dtype": "float32"
23
+ }