chore(impute_missing): use coarser granularity: 1min -> 5min windows

OxWearables · Apr 26, 2024 · 4f8913a · 4f8913a
1 parent 8ae69cf
commit 4f8913a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/stepcount/stepcount.py b/src/stepcount/stepcount.py
@@ -423,13 +423,13 @@ def fillna(subframe):
     data = (
         data
         # first attempt imputation using same day of week
-        .groupby([data.index.weekday, data.index.hour, data.index.minute])
+        .groupby([data.index.weekday, data.index.hour, data.index.minute // 5])
         .transform(fillna)
         # then try within weekday/weekend
-        .groupby([data.index.weekday >= 5, data.index.hour, data.index.minute])
+        .groupby([data.index.weekday >= 5, data.index.hour, data.index.minute // 5])
         .transform(fillna)
         # finally, use all other days
-        .groupby([data.index.hour, data.index.minute])
+        .groupby([data.index.hour, data.index.minute // 5])
         .transform(fillna)
     )