From bdcb7dbd8194cdb098e4f9737c1b00123c899ab9 Mon Sep 17 00:00:00 2001 From: simGPT Date: Tue, 2 Jun 2026 15:49:39 +0900 Subject: [PATCH] =?UTF-8?q?:bug:Fix:=20churn=20loader=20=EA=B5=AC=EB=B2=84?= =?UTF-8?q?=EC=A0=84=20=ED=95=A8=EC=88=98=20=EC=A0=9C=EA=B1=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- services/model-api/app/models/loader.py | 15 --------------- 1 file changed, 15 deletions(-) diff --git a/services/model-api/app/models/loader.py b/services/model-api/app/models/loader.py index d515281..02e6174 100644 --- a/services/model-api/app/models/loader.py +++ b/services/model-api/app/models/loader.py @@ -35,18 +35,3 @@ def load_churn_model(model_name: str): _model_cache[cache_key] = model return model - -# mlflow에서 고객 이탈 예측 모델 로드하는 함수 -def load_churn_model(model_name: str, version: str): - cache_key = f"{model_name}_{version}_sklearn" - if cache_key in _model_cache: - return _model_cache[cache_key] - - mlflow_uri = os.getenv("MLFLOW_TRACKING_URI", "http://mlflow:5000") - mlflow.set_tracking_uri(mlflow_uri) - - model_uri = f"models:/{model_name}-{version}/latest" - model = mlflow.sklearn.load_model(model_uri) - - _model_cache[cache_key] = model - return model