leuschnm commited on
Commit
31f456a
·
1 Parent(s): a852418
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -38,7 +38,7 @@ def raw_preds_to_df(raw,quantiles = None):
38
  preds_df['pred_idx'] = preds_df['time_idx'] + preds_df['h'] - 1
39
  return preds_df
40
 
41
- @st.cache_data
42
  def prepare_dataset(_parameters, df, rain, temperature, datepicker, mapping):
43
  if rain != "Default":
44
  df["MTXWTH_Day_precip"] = mapping[rain]
@@ -54,13 +54,13 @@ def prepare_dataset(_parameters, df, rain, temperature, datepicker, mapping):
54
  df = TimeSeriesDataSet.from_parameters(_parameters, df)
55
  return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
56
 
57
- @st.cache_data
58
  def predict(_model, _dataloader, datepicker):
59
  out = _model.predict(_dataloader, mode="raw", return_x=True, return_index=True)
60
  preds = raw_preds_to_df(out, quantiles = None)
61
  return preds[["pred_idx", "Group", "pred"]]
62
 
63
- @st.cache_data
64
  def adjust_data_for_plot(df, preds):
65
  df = pd.merge(df, preds, left_on=["time_idx", "Group"], right_on=["pred_idx", "Group"], how = "left")
66
  df = df[~df["pred"].isna()]
 
38
  preds_df['pred_idx'] = preds_df['time_idx'] + preds_df['h'] - 1
39
  return preds_df
40
 
41
+ #@st.cache_data
42
  def prepare_dataset(_parameters, df, rain, temperature, datepicker, mapping):
43
  if rain != "Default":
44
  df["MTXWTH_Day_precip"] = mapping[rain]
 
54
  df = TimeSeriesDataSet.from_parameters(_parameters, df)
55
  return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
56
 
57
+ #@st.cache_data
58
  def predict(_model, _dataloader, datepicker):
59
  out = _model.predict(_dataloader, mode="raw", return_x=True, return_index=True)
60
  preds = raw_preds_to_df(out, quantiles = None)
61
  return preds[["pred_idx", "Group", "pred"]]
62
 
63
+ #@st.cache_data
64
  def adjust_data_for_plot(df, preds):
65
  df = pd.merge(df, preds, left_on=["time_idx", "Group"], right_on=["pred_idx", "Group"], how = "left")
66
  df = df[~df["pred"].isna()]