Hugging Face's logo

Spaces:
leuschnm
/
TemporalFusionTransformer
Runtime error

App Files Community
leuschnm commited on
Commit
9f8b254
·
1 Parent(s): 83bf3df

bug fix

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ def prepare_dataset(parameters, df, rain, temperature, datepicker, mapping):
57
  def predict(_model, _dataloader, datepicker):
58
  out = _model.predict(_dataloader, mode="raw", return_x=True, return_index=True)
59
  preds = raw_preds_to_df(out, quantiles = None)
60
- return preds[["pred_idx", "Group", "pred"]
61
 
62
 
63
  def generate_plot(df, preds):
 
57
  def predict(_model, _dataloader, datepicker):
58
  out = _model.predict(_dataloader, mode="raw", return_x=True, return_index=True)
59
  preds = raw_preds_to_df(out, quantiles = None)
60
+ return preds[["pred_idx", "Group", "pred"]]
61
 
62
 
63
  def generate_plot(df, preds):