mlflow.models

class mlflow.models.Model(artifact_path=None, run_id=None, utc_time_created=datetime.datetime(2018, 6, 27, 15, 2, 55, 828478), flavors=None)

Bases: object

A servable MLflow model, which can support multiple model flavors.

add_flavor(name, **params)

Add an entry for how to serve the model in a given format.

classmethod load(path)

Load a Servable from its YAML representation.

save(path)

Write this Servable as a YAML file to a local file.

to_yaml(stream=None)