Spaces:
Runtime error
Runtime error
bug fix
Browse files
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 |
-
|
| 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 |
-
|
| 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 |
-
|
| 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()]
|