MLflow Google Cloud Vertex AI integration package
Project description
MLflow plugin for Google Cloud Vertex AI
Installation
python3 -m pip install google_cloud_mlflow
Deployment plugin usage
Command-line
Create deployment
mlflow deployments create --target google_cloud --name "deployment name" --model-uri "models:/mymodel/mymodelversion" --config destination_image_uri="gcr.io/<repo>/<path>"
List deployments
mlflow deployments list --target google_cloud
Get deployment
mlflow deployments get --target google_cloud --name "deployment name"
Delete deployment
mlflow deployments delete --target google_cloud --name "deployment name"
Update deployment
mlflow deployments update --target google_cloud --name "deployment name" --model-uri "models:/mymodel/mymodelversion" --config destination_image_uri="gcr.io/<repo>/<path>"
Predict
mlflow deployments predict --target google_cloud --name "deployment name" --input-path "inputs.json" --output-path "outputs.json
Get help
mlflow deployments help --target google_cloud
Python
from mlflow import deployments
client = deployments.get_deploy_client("google_cloud")
# Create deployment
model_uri = "models:/mymodel/mymodelversion"
deployment = client.create_deployment(
name="deployment name",
model_uri=model_uri,
# Config is optional
config=dict(
# Deployed model config
machine_type="n1-standard-2",
min_replica_count=None,
max_replica_count=None,
accelerator_type=None,
accelerator_count=None,
service_account=None,
explanation_metadata=None, # JSON string
explanation_parameters=None, # JSON string
# Model container image building config
destination_image_uri=None,
timeout=None,
# Model deployment config
sync="true",
# Endpoint config
description=None,
# Vertex AI config
project=None,
location=None,
experiment=None,
experiment_description=None,
staging_bucket=None,
# List deployments
deployments = client.list_deployments()
# Get deployment
deployments = client.get_deployment(name="deployment name")
# Delete deployment
deployment = client.delete_deployment(name="deployment name")
# Update deployment
deployment = client.create_deployment(
name="deployment name",
model_uri=model_uri,
# Config is optional
config=dict(...),
)
# Predict
import pandas
df = pandas.DataFrame([
{"a": 1,"b": 2,"c": 3},
{"a": 4,"b": 5,"c": 6}
])
predictions = client.predict("deployment name", df)
Model Registry plugin usage
Set the MLflow Model Registry URI to a directory in some Google Cloud Storage bucket, then log models using mlflow.log_model
as usual.
mlflow.set_registry_uri("gs://<bucket>/models/")
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for google_cloud_mlflow-0.0.4.dev2.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 77d440daaf844254669cef4b38942c304c1dd9c638251770d39df43a1e57efa3 |
|
MD5 | 05ac37c6fba19bd6a4ec9e7596b4683d |
|
BLAKE2b-256 | 736e2a4655acb54bffd13184a9e035ad933d90a4f77f618605b9581d7ac689fe |
Close
Hashes for google_cloud_mlflow-0.0.4.dev2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e1333bc5b2c1c772d0528208eb059c8c6618f90ed37fe8de6bc42701b293def0 |
|
MD5 | a8e56a8a3df0ec0fe7104942413e0350 |
|
BLAKE2b-256 | 0bb61f000626eb2910ffe3f92ea5af71bbd025810963f7dfe2ad215fccad1922 |