Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -220,7 +220,7 @@ def select_model(horizon, model_type, max_steps=50):
|
|
220 |
max_steps=max_steps)
|
221 |
elif model_type == 'TFT':
|
222 |
return TFT(h=horizon,
|
223 |
-
input_size=horizon,
|
224 |
hidden_size=16,
|
225 |
loss=HuberMQLoss(level=[90]),
|
226 |
learning_rate=0.005,
|
|
|
220 |
max_steps=max_steps)
|
221 |
elif model_type == 'TFT':
|
222 |
return TFT(h=horizon,
|
223 |
+
input_size=horizon*5,
|
224 |
hidden_size=16,
|
225 |
loss=HuberMQLoss(level=[90]),
|
226 |
learning_rate=0.005,
|