Spaces:

leuschnm
/

TemporalFusionTransformer

Runtime error

leuschnm commited on Jul 2, 2023

Commit

b635c73

1 Parent(s): 9f8b254

bug fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -93,6 +93,7 @@ def load_data():
         parameters = pickle.load(f)
     df = pd.read_pickle('data/test_data.pkl')
     df = df.loc[(df["Branch"] == "15") & (df["Group"].isin(["6","7","4","1"]))]
     return parameters, df
 @st.cache_resource

         parameters = pickle.load(f)
     df = pd.read_pickle('data/test_data.pkl')
     df = df.loc[(df["Branch"] == "15") & (df["Group"].isin(["6","7","4","1"]))]
+    df = df.replace(0, np.nan)
     return parameters, df
 @st.cache_resource