Надання моделі в сервіс

Вступ до MLflow

Weston Bassler

Senior MLOps Engineer

MLflow Models

Стандартизувати пакування моделей
Логувати моделі
Оцінювання моделей

Розгортання моделі

життєвий цикл ML

¹ datacamp.com

REST API

/ping — для перевірок стану
/health — для перевірок стану
/version — щоб отримати версію MLflow
/invocations — для скорингу моделі
Порт 5000

Кінцева точка Invocations

/invocations

No,Name,Subject
1,Bill Johnson,English
2,Gary Valentine,Mathematics

Content-Type:application/json або application/csv

{
    "1": {
        "No": "1",
        "Name": "Bill Johnson",
        "Subject": "English"
    },
    "2": {
        "No": "2",
        "Name": "Gary Valentine",
        "Subject": "Mathematics"
    }
}

Формат CSV

Датафрейм pandas
pandas_df.to_csv()

Формат JSON

dataframe_split — pandas DataFrame у поданні split
dataframe_records — pandas DataFrame у поданні records

DataFrame split

# Подання DataFrame split
{
  "dataframe_split": {
      "columns": ["sex", "age", "weight"],
      "data": [["male", 23, 160], ["female", 33, 120]]
  }
}

Надання моделей у сервіс

# Команда serve для MLflow
mlflow models serve --help
Usage: mlflow models serve [OPTIONS]

URI для serve

# Локальна файлова система
mlflow models serve -m relative/path/to/local/model

# Run ID
mlflow models serve -m runs:/<mlflow_run_id>/artifacts/model

# AWS S3
mlflow models serve -m s3://my_bucket/path/to/model

Приклад serve

# Надати модель із run
mlflow models serve -m runs:/e84a122920de4bdeaedb54146deeb429/artifacts/model

2023/03/12 16:28:28 INFO mlflow.models.flavor_backend_registry: 
Selected backend for flavor 'python_function'
2023/03/12 16:28:28 INFO mlflow.pyfunc.backend: === Running command 
'exec gunicorn --timeout=60 -b 127.0.0.1:5000 -w 1 ${GUNICORN_CMD_ARGS} -- 
mlflow.pyfunc.scoring_server.wsgi:app'
[2023-03-12 16:28:29 -0400] [48431] [INFO] Starting gunicorn 20.1.0
[2023-03-12 16:28:29 -0400] [48431] [INFO] Listening at: http://127.0.0.1:5000 
(48431)
[2023-03-12 16:28:29 -0400] [48431] [INFO] Using worker: sync
[2023-03-12 16:28:29 -0400] [48432] [INFO] Booting worker with pid: 48432

Запит до Invocations

# Надіслати корисне навантаження у форматі dataframe_split до MLflow
curl http://127.0.0.1:5000/invocations -H 'Content-Type: application/json' -d '{
  "dataframe_split": {
      "columns": ["sex", "age", "weight"],
      "data": [["male", 23, 160], ["female", 33, 120]]
  }
}'

{"predictions": [1, 0]}

Давайте потренуємось!

Вступ до MLflow