leuschnm commited on
Commit
adb8bdc
·
1 Parent(s): cf70ab7

change conversion

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, return_index=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"]]