Spaces:
Runtime error
Runtime error
change conversion
Browse files
app.py
CHANGED
|
@@ -58,7 +58,7 @@ def prepare_dataset(_parameters, df, rain, temperature, datepicker, mapping):
|
|
| 58 |
return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
|
| 59 |
|
| 60 |
def predict(_model, _dataloader, datepicker, df):
|
| 61 |
-
out = _model.predict(_dataloader, mode="raw", 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"]]
|
|
|
|
| 58 |
return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
|
| 59 |
|
| 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"]]
|