fix(classification): skip model predict if input all NA

OxWearables · Nov 6, 2024 · ccf4f04 · ccf4f04
1 parent ba14445
commit ccf4f04
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/accelerometer/classification.py b/src/accelerometer/classification.py
@@ -55,7 +55,8 @@ def activityClassification(
     print(f"{len(epoch) - np.sum(ok)} rows with NaN or Inf values, out of {len(epoch)}")
 
     Y = pd.Series(index=epoch.index)
-    Y.loc[ok] = viterbi(model.predict(X[ok]), hmmParams)
+    if ok.any():
+        Y.loc[ok] = viterbi(model.predict(X[ok]), hmmParams)
 
     # TODO: Chan's logic hardcoded here
     if activityModel == 'chan':