Spaces:
Runtime error
Runtime error
change conversion
Browse files
app.py
CHANGED
|
@@ -60,7 +60,7 @@ def prepare_dataset(_parameters, df, rain, temperature, datepicker, mapping):
|
|
| 60 |
def predict(_model, _dataloader, datepicker, df):
|
| 61 |
out = _model.predict(_dataloader, mode="raw", return_index=True)# return_x=True,
|
| 62 |
first_date = min(df["Date"])
|
| 63 |
-
preds = raw_preds_to_df(out, first_date)
|
| 64 |
return preds[["pred_idx", "Group", "pred"]]
|
| 65 |
|
| 66 |
def adjust_data_for_plot(df, preds):
|
|
|
|
| 60 |
def predict(_model, _dataloader, datepicker, df):
|
| 61 |
out = _model.predict(_dataloader, mode="raw", return_index=True)# return_x=True,
|
| 62 |
first_date = min(df["Date"])
|
| 63 |
+
preds = raw_preds_to_df(raw = out, idx_offset = first_date)
|
| 64 |
return preds[["pred_idx", "Group", "pred"]]
|
| 65 |
|
| 66 |
def adjust_data_for_plot(df, preds):
|