From 725a4349eea58dea5c7fb15a1cff6c969fc5092e Mon Sep 17 00:00:00 2001
From: 21in7 <skawksmsduzo@gmail.com>
Date: Mon, 2 Mar 2026 01:08:12 +0900
Subject: [PATCH] chore: Update MLXFilter model deployment and logging with new
 training results and ONNX file management

- Added new training log entries for lgbm backend with AUC, precision, and recall metrics.
- Enhanced deploy_model.sh to manage ONNX and lgbm model files based on the selected backend.
- Adjusted output shape in mlx_filter.py for ONNX export to support dynamic batch sizes.
---
 models/mlx_filter.weights.onnx | Bin 45243 -> 45246 bytes
 models/training_log.json       |  24 ++++++++++++++++++++++++
 scripts/deploy_model.sh        |  21 +++++++++++++++++----
 src/mlx_filter.py              |   2 +-
 4 files changed, 42 insertions(+), 5 deletions(-)

diff --git a/models/mlx_filter.weights.onnx b/models/mlx_filter.weights.onnx
index 1856b62b36f99499311a65b886a94a25fe0fe283..d0800fd0e500c462072b37ae418af3b9e6eb11a6 100644
GIT binary patch
delta 43
zcmdn}kZIpTCKe73t4$kO=5LS|;bJW)%1=rZ;^yMyU=(8IViscbb7J9Q5D)|aB8&;e

delta 40
wcmdn@kZJcrCKe73tMwaM=5LS?<YFx-%1=rZ;^1QAU=(8FVsK*NVh|7n00VysPXGV_

diff --git a/models/training_log.json b/models/training_log.json
index 5877a0d..8d55ddc 100644
--- a/models/training_log.json
+++ b/models/training_log.json
@@ -240,5 +240,29 @@
     "train_sec": 0.2,
     "time_weight_decay": 2.0,
     "model_path": "models/mlx_filter.weights"
+  },
+  {
+    "date": "2026-03-02T00:54:32.264425",
+    "backend": "lgbm",
+    "auc": 0.5607,
+    "best_threshold": 0.6532,
+    "best_precision": 0.467,
+    "best_recall": 0.2,
+    "samples": 533,
+    "features": 23,
+    "time_weight_decay": 2.0,
+    "model_path": "models/lgbm_filter.pkl"
+  },
+  {
+    "date": "2026-03-02T01:07:30.690959",
+    "backend": "lgbm",
+    "auc": 0.5579,
+    "best_threshold": 0.6511,
+    "best_precision": 0.4,
+    "best_recall": 0.171,
+    "samples": 533,
+    "features": 23,
+    "time_weight_decay": 2.0,
+    "model_path": "models/lgbm_filter.pkl"
   }
 ]
\ No newline at end of file
diff --git a/scripts/deploy_model.sh b/scripts/deploy_model.sh
index be8ff35..143d4f8 100755
--- a/scripts/deploy_model.sh
+++ b/scripts/deploy_model.sh
@@ -33,12 +33,25 @@ done
 echo "=== 모델 전송 시작 (백엔드: ${BACKEND}) ==="
 echo "  대상: ${LXC_HOST}:${LXC_MODELS_PATH}"
 
-# ── 원격 디렉터리 생성 + lgbm 기존 모델 백업 ─────────────────────────────────
+# ── 원격 디렉터리 생성 + 백업 + 상대 백엔드 파일 제거 ───────────────────────
+# lgbm 배포 시: 기존 lgbm 백업 후 ONNX 파일 삭제 (ONNX 우선순위 때문에 lgbm이 무시되는 것 방지)
+# mlx 배포 시: lgbm 파일 삭제 (명시적으로 mlx만 사용)
 ssh "${LXC_HOST}" "
   mkdir -p '${LXC_MODELS_PATH}'
-  if [ '$BACKEND' = 'lgbm' ] && [ -f '${LXC_MODELS_PATH}/lgbm_filter.pkl' ]; then
-    cp '${LXC_MODELS_PATH}/lgbm_filter.pkl' '${LXC_MODELS_PATH}/lgbm_filter_prev.pkl'
-    echo '  기존 lgbm 모델 백업 완료'
+  if [ '$BACKEND' = 'lgbm' ]; then
+    if [ -f '${LXC_MODELS_PATH}/lgbm_filter.pkl' ]; then
+      cp '${LXC_MODELS_PATH}/lgbm_filter.pkl' '${LXC_MODELS_PATH}/lgbm_filter_prev.pkl'
+      echo '  기존 lgbm 모델 백업 완료'
+    fi
+    if [ -f '${LXC_MODELS_PATH}/mlx_filter.weights.onnx' ]; then
+      rm '${LXC_MODELS_PATH}/mlx_filter.weights.onnx'
+      echo '  ONNX 파일 제거 완료 (lgbm 우선 적용)'
+    fi
+  else
+    if [ -f '${LXC_MODELS_PATH}/lgbm_filter.pkl' ]; then
+      rm '${LXC_MODELS_PATH}/lgbm_filter.pkl'
+      echo '  lgbm 파일 제거 완료 (mlx 우선 적용)'
+    fi
   fi
 "
 
diff --git a/src/mlx_filter.py b/src/mlx_filter.py
index 8902b0e..639a914 100644
--- a/src/mlx_filter.py
+++ b/src/mlx_filter.py
@@ -84,7 +84,7 @@ def _export_onnx(
         nodes,
         "mlx_filter",
         inputs=[helper.make_tensor_value_info("X", TensorProto.FLOAT, [None, input_dim])],
-        outputs=[helper.make_tensor_value_info("proba", TensorProto.FLOAT, [None])],
+        outputs=[helper.make_tensor_value_info("proba", TensorProto.FLOAT, [-1])],
         initializer=initializers,
     )
     model_proto = helper.make_model(graph, opset_imports=[helper.make_opsetid("", 17)])