Shoriful025 commited on
Commit
e1426eb
·
verified ·
1 Parent(s): e9d076d

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +26 -0
config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "TimeSeriesTransformerForPrediction"
4
+ ],
5
+ "context_length": 30,
6
+ "d_model": 64,
7
+ "decoder_attention_heads": 4,
8
+ "decoder_ffn_dim": 128,
9
+ "decoder_layers": 3,
10
+ "distribution_output": "student_t",
11
+ "dropout": 0.1,
12
+ "encoder_attention_heads": 4,
13
+ "encoder_ffn_dim": 128,
14
+ "encoder_layers": 3,
15
+ "feature_size": 18,
16
+ "input_size": 1,
17
+ "lags_sequence": [1, 7, 30],
18
+ "loss": "nll",
19
+ "model_type": "time_series_transformer",
20
+ "num_dynamic_real_features": 6,
21
+ "num_static_categorical_features": 1,
22
+ "prediction_length": 15,
23
+ "scaling": "mean",
24
+ "torch_dtype": "float32",
25
+ "transformers_version": "4.36.0"
26
+ }