Úvod do MLflow Projects

Úvod do MLflow

Weston Bassler

Senior MLOps Engineer

MLflow Projects

  • Reprodukovatelné

  • Znovupoužitelné

  • Přenositelné

Zvýšení produktivity

1 unsplash.com
Úvod do MLflow

MLproject

project/
    MLproject
    train_model.py
    python_env.yaml
    requirements.txt

Repozitář Github Git

1 github.com
Úvod do MLflow

Soubor MLproject

  • name: - Název projektu

  • entry_points:

    • Příkaz(y) ke spuštění
    • Soubory .py a .sh
    • Pracovní postupy

  • python_env:

    • Prostředí Pythonu
    • python_env.yaml

YAML

1 wikipedia.org
Úvod do MLflow

Příklad souboru MLproject

name: salary_model

entry_points:
  main:
    command: "python train_model.py"

python_env: python_env.yaml
Úvod do MLflow

train_model.py

# Import libraries and modules
import mlflow
import mlflow.sklearn
import pandas as pd
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import train_test_split


# Training Data
df = pd.read_csv('Salary_predict.csv')
X = df[["experience", "age", "interview_score"]]
y = df[["Salary"]]
X_train, X_test, y_train, y_test = train_test_split(X, y, train_size=0.7,random_state=0)
Úvod do MLflow

train_model.py

# Set Auto logging for Scikit-learn flavor
mlflow.sklearn.autolog()


# Train the model
lr = LinearRegression()
lr.fit(X_train, y_train)
Úvod do MLflow

python_env.yaml

python: 3.10.8

build_dependencies:
- pip
- setuptools
- wheel

dependencies:
- -r requirements.txt
Úvod do MLflow

requirements.txt

mlflow
scikit-learn
Úvod do MLflow

Pojďme procvičovat!

Úvod do MLflow

Preparing Video For Download...