Spaces:
Runtime error
Runtime error
bug fix
Browse files
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 |
-
|
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 |
|