iris_sklearn_svc

Iris 데이터셋의 SVM 분류 모델 학습 파이프라인

목적

MLFlow 를 사용한 Iris 데이터셋의 SVM 분류 모델 학습 파이프라인입니다.

전제

• Python 3.8 이상
• Docker
• MLFlow

사용법

0. 현재 디렉토리

$ pwd
~/ml-system-in-actions/chapter2_training/iris_sklearn_svc

1. 라이브러리 인스톨 이 프로그램에서 사용하는 라이브러리는requirements.txt에 기재되어 있습니다.

$ make dev
# 실행 커맨드
# pip install -r requirements.txt
# 출력 생략

2. 학습용 Docker 이미지 빌드 학습에서 사용할 Docker 이미지를 빌드합니다.

$ make d_build
# 실행 커맨드
# docker build \
#     -t shibui/ml-system-in-actions:training_pattern_iris_sklearn_svc_0.0.1 \
#     -f Dockerfile .
# 출력 생략
# docker 이미지로 shibui/ml-system-in-actions:training_pattern_iris_sklearn_svc_0.0.1 이 빌드됩니다.

3. 학습 파이프라인 실행 mlflow 로 학습 파이프라인을 실행합니다.

$ make train
# 실행 커맨드
# mlflow run . --no-conda
# 출력 예
# 2021/02/11 11:29:41 INFO mlflow.projects.docker: === Building docker image iris_sklearn_svc:6fa928e ===
# 2021/02/11 11:29:52 INFO mlflow.projects.utils: === Created directory /var/folders/v8/bvkzgn8j1ws6y76t4z5nt6280000gn/T/tmpdtqjaxok for downloading remote URIs passed to arguments of type 'path' ===
# 2021/02/11 11:29:52 INFO mlflow.projects.backend.local: === Running command 'docker run --rm -v ~/book/ml-system-in-actions/chapter2_training/iris_sklearn_svc/mlruns:/mlflow/tmp/mlruns -v ~/book/ml-system-in-actions/chapter2_training_patterns/iris_sklearn_svc/mlruns/0/c2e44e8158f64352b4f23fb21329693e/artifacts:~/book/ml-system-in-actions/chapter2_training_patterns/iris_sklearn_svc/mlruns/0/c2e44e8158f64352b4f23fb21329693e/artifacts -e MLFLOW_RUN_ID=c2e44e8158f64352b4f23fb21329693e -e MLFLOW_TRACKING_URI=file:///mlflow/tmp/mlruns -e MLFLOW_EXPERIMENT_ID=0 iris_sklearn_svc:6fa928e python -m iris_train \
#   --test_size 0.3 ' in run with ID 'c2e44e8158f64352b4f23fb21329693e' ===
# 2021/02/11 11:29:58 INFO mlflow.projects: === Run (ID 'c2e44e8158f64352b4f23fb21329693e') succeeded ===

학습은 수분 이내로 완료됩니다.