From 6b95c617348f6ae6a6acfa1acda51189442ea48b Mon Sep 17 00:00:00 2001 From: Evan Ray Date: Tue, 19 Nov 2024 16:53:12 -0500 Subject: [PATCH] Update sarix.py to address missing values in input nhsn data --- src/idmodels/sarix.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/idmodels/sarix.py b/src/idmodels/sarix.py index 6bd885e..5a94a87 100644 --- a/src/idmodels/sarix.py +++ b/src/idmodels/sarix.py @@ -30,6 +30,7 @@ def run(self, run_config): df["xmas_spike"] = np.maximum(3 - np.abs(df["delta_xmas"]), 0) xy_colnames = ["inc_trans_cs"] + self.model_config.x + df = df.query("wk_end_date >= '2022-10-01'").interpolate() batched_xy = df[xy_colnames].values.reshape(len(df["location"].unique()), -1, len(xy_colnames)) sarix_fit_all_locs_theta_pooled = sarix.SARIX(