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()]
|