Shoriful025 commited on
Commit
97bd1cd
·
verified ·
1 Parent(s): 41f4f9c

Create config.json

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