Introductie tot MLflow
Weston Bassler
Senior MLOps Engineer

Modelversies
Modelstages

Model laden
# MLflow-flavor
mlflow.FLAVOR.load_model()
Model serveren
# MLflow-serve-opdrachtregel
mlflow models serve
Conventie
models:/
Modelversie
models:/model_name/version
Modelstage
models:/model_name/stage
# Flavor importeren import mlflow.FLAVOR# Versie laden mlflow.FLAVOR.load_model("models:/model_name/version")# Stage laden mlflow.FLAVOR.load_model("models:/model_name/stage")
# Flavor importeren import mlflow.sklearn# Unicorn-model in Staging laden model = mlflow.sklearn.load_model("models:/Unicorn/Staging")# Model afdrukken model
LogisticRegression()
# Inference
model.predict(data)
# Unicorn-model serveren in Production-stage
mlflow models serve -m "models:/Unicorn/Production"
2023/03/26 15:07:00 INFO mlflow.models.flavor_backend_registry:
Selected backend for flavor 'python_function'
2023/03/26 15:07:00 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-26 15:07:00 -0400] [86409] [INFO] Starting gunicorn 20.1.0
[2023-03-26 15:07:00 -0400] [86409] [INFO] Listening at: http://127.0.0.1:5000
[2023-03-26 15:07:00 -0400] [86409] [INFO] Using worker: sync
[2023-03-26 15:07:00 -0400] [86410] [INFO] Booting worker with pid: 86410
CSV-indeling
pandas_df.to_csv()
JSON-indeling
{
"dataframe_split": {
"columns": ["R&D Spend", "Administration", "Marketing Spend", "State"],
"data": [["165349.20", 136897.80, 471784.10, 1]]
}
}
# Payload naar het invocations-endpoint sturen
curl http://127.0.0.1:5000/invocations -H 'Content-Type: application/json' -d
{
"dataframe_split": {
"columns": ["R&D Spend", "Administration", "Marketing Spend", "State"],
"data": [["165349.20", 136897.80, 471784.10, 1]]
}
}
[[104055.1842384]]
Introductie tot MLflow