| { | |
| "_name_or_path": "EcomSalesTrendPredictor", | |
| "architectures": ["TimeSeriesTransformerModel"], | |
| "model_type": "time_series_transformer", | |
| "prediction_length": 7, | |
| "context_length": 30, | |
| "scaling": "std", | |
| "input_size": 1, | |
| "num_layers": 3, | |
| "hidden_size": 64, | |
| "num_heads": 4, | |
| "output_type": "quantiles", | |
| "lags_sequence": [1, 2, 7, 30], | |
| "num_static_real_features": 3, | |
| "num_dynamic_real_features": 4, | |
| "num_static_categorical_features": 4, | |
| "num_dynamic_categorical_features": 3, | |
| "feat_static_cat_cardinalities": [4, 4, 2, 2], | |
| "feat_dynamic_cat_cardinalities": [4, 4, 2], | |
| "dropout": 0.1, | |
| "quantiles": [0.1, 0.5, 0.9], | |
| "torch_dtype": "float32" | |
| } |