diff --git a/models/mlx_filter.weights.onnx b/models/mlx_filter.weights.onnx index 1856b62..d0800fd 100644 Binary files a/models/mlx_filter.weights.onnx and b/models/mlx_filter.weights.onnx differ 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)])