Tasfiya025's picture
Create config.json
ba791d6 verified
{
"model_type": "time_series_transformer",
"architectures": ["TimeSeriesTransformerForPrediction"],
"prediction_length": 24,
"context_length": 168,
"input_size": 6,
"num_time_features": 4,
"hidden_size": 256,
"num_attention_heads": 8,
"num_hidden_layers": 6,
"dropout": 0.1
}