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"]]
|