feat: support multivariate threshold output (#314)

Multivariate thresholding produces a single-dimensional output. This change makes sure that we preserve the dimensions in postprocess in order to create a valid OutputPayload Signed-off-by: Avik Basu <[email protected]>
numaproj · Oct 12, 2023 · 509e38a · 509e38a
1 parent 64c2e95
commit 509e38a
Show file tree

Hide file tree

Showing 4 changed files with 898 additions and 833 deletions.
diff --git a/numalogic/udfs/postprocess.py b/numalogic/udfs/postprocess.py
@@ -191,7 +191,7 @@ def compute(
  except Exception as err:
  raise RuntimeError("Threshold model scoring failed") from err
  try:
- win_score = np.mean(y_score, axis=0)
+ win_score = np.mean(y_score, axis=0, keepdims=True)
  score = postproc_clf.transform(win_score)
  except Exception as err:
  raise RuntimeError("Postprocess failed") from err