mlflow.models

class mlflow.models.Model(artifact_path=None, run_id=None, utc_time_created=datetime.datetime(2018, 8, 8, 1, 43, 49, 951572), flavors=None)

Bases: object

A MLflow model that 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 model from its YAML representation.

classmethod log(artifact_path, flavor, **kwargs)

Log model using supplied flavor module.

Parameters:
  • artifact_path – Run relative path identifying this model.
  • flavor – Flavor module / object to save the model with. The module / object must have the save_model function that will persist the model as a valid MLflow model.
  • kwargs – Extra args passed to the model flavor.
save(path)

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

to_yaml(stream=None)