[MLOps] MLflow 실습 - 4

GisangLee·2023년 3월 25일
0

ML

목록 보기
133/141

1. 다른 예제 살펴보기

$ wget https://raw.githubusercontent.com/mlflow/mlflow/master/examples/sklearn_autolog/utils.py
$ wget https://raw.githubusercontent.com/mlflow/mlflow/master/examples/sklearn_autolog/pipeline.py
  • mlflow에서 예제로 제공해주는 example 중 하나

  • 간단한 training data를 가지고 sklearnpipeline을 사용해,
    StandardScaler 전처리 이후 Linear Regression을 수행하는 코드

  • scikit-learn과 같은 패키지는 mlflow 레벨에서 autolog를 지원

    • model의 parameters, metrics와 model artifacts를 사용자가
      명시하지 않아도 자동으로 mlflow에 로깅

2. 코드 살펴보기 (XGB Model)

$ wget https://raw.githubusercontent.com/mlflow/mlflow/master/examples/xgboost/train.py

# xgboost==1.4.2 설치 필요
profile
포폴 및 이력서 : https://gisanglee.github.io/web-porfolio/

0개의 댓글