Skip to content

[Refactor]: MLflow 모델 버전 관리 방식 개선 #36

Description

@simGPT

🧐 리팩토링 대상 및 이유

현재 모델명에 버전을 직접 포함(churn-v1, churn-v2)하는 방식 대신, MLflow 내장 버전 관리를 활용하여 모델명을 churn으로 통일하고 버전은 MLflow가 자동 관리하도록 변경합니다.

🛠️ 작업 상세 내용

  • train.py registered_model_name → churn으로 변경
  • loader.py 모델 로드 방식 MLflow 버전 기반으로 변경
  • deployment.yaml CHURN_MODEL_VERSION 환경변수 제거 또는 변경
  • EC2 MLflow에 모델 재등록

✅ 확인 사항

  • 기존 기능이 정상적으로 작동함을 확인했나요?
  • 테스트 코드가 있다면 테스트를 통과했나요?

Metadata

Metadata

Assignees

No one assigned

    Labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions