azrai99 commited on
Commit
5f28cd3
·
verified ·
1 Parent(s): c0debe7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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,