Upload TimeSeriesTransformerForPrediction

config.json

@@ -9,7 +9,7 @@
     366
   ],
   "context_length": 24,
-  "d_model": 24,
+  "d_model": 26,
   "decoder_attention_heads": 2,
   "decoder_ffn_dim": 32,
   "decoder_layerdrop": 0.1,
@@ -17,7 +17,7 @@
   "distribution_output": "student_t",
   "dropout": 0.1,
   "embedding_dimension": [
-    4
+    6
   ],
   "encoder_attention_heads": 2,
   "encoder_ffn_dim": 32,
@@ -26,7 +26,7 @@
   "init_std": 0.02,
   "input_size": 1,
   "is_encoder_decoder": true,
-  "lags_seq": [
+  "lags_sequence": [
     1,
     2,
     3,
@@ -46,10 +46,10 @@
   ],
   "loss": "nll",
   "model_type": "time_series_transformer",
-  "num_feat_dynamic_real": 0,
-  "num_feat_static_cat": 1,
-  "num_feat_static_real": 0,
+  "num_dynamic_real_features": 0,
   "num_parallel_samples": 100,
+  "num_static_categorical_features": 1,
+  "num_static_real_features": 0,
   "num_time_features": 2,
   "prediction_length": 24,
   "scaling": true,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bf71022d107ea2dabe9670300b773eb44eaaa337cea99ed1d033743e534e6b39
-size 119557
+oid sha256:3fadbd056a94e13bb37071615b2415037197d5d203eb7d76e84b530e10cbc99d
+size 133829