leuschnm commited on
Commit
61e3a2e
·
1 Parent(s): f0fac46
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -57,9 +57,11 @@ 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
 
61
- date_list = [datepicker + datetime.timedelta(days=x) for x in range(30)]
62
- preds["date_imputed"] = date_list
63
 
64
  return preds[["date_imputed", "Group", "pred"]]
65
 
 
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
+
61
+ def add_dates(group, datepicker):
62
+ return group["date_imputed"] = [datepicker + datetime.timedelta(days=x) for x in range(30)]
63
 
64
+ preds["date_imputed"] = preds.groupby("Group").transform(add_dates)
 
65
 
66
  return preds[["date_imputed", "Group", "pred"]]
67