fix: take mean before calculating the thresholds (#340)

take mean before calculating the thresholds --------- Signed-off-by: s0nicboOm <[email protected]>
numaproj · Jan 20, 2024 · 870f263 · 870f263
1 parent dbb510f
commit 870f263
Show file tree

Hide file tree

Showing 2 changed files with 572 additions and 536 deletions.
diff --git a/numalogic/udfs/postprocess.py b/numalogic/udfs/postprocess.py
@@ -213,12 +213,12 @@ def compute(
         """
         _start_time = time.perf_counter()
         try:
-            y_score = model.score_samples(input_).astype(np.float32)
+            score = np.mean(input_, axis=0, keepdims=True)
+            y_score = model.score_samples(score).astype(np.float32)
         except Exception as err:
             raise RuntimeError("Threshold model scoring failed") from err
         try:
-            win_score = np.mean(y_score, axis=0, keepdims=True)
-            score = postproc_clf.transform(win_score)
+            score = postproc_clf.transform(y_score)
         except Exception as err:
             raise RuntimeError("Postprocess failed") from err
         _LOGGER.debug(