mlflow (version 1.9.0)

mlflow_save_model.crate: Save Model for MLflow

Description

Saves model in MLflow format that can later be used for prediction and serving. This method is generic to allow package authors to save custom model types.

Usage

# S3 method for crate
mlflow_save_model(model, path, model_spec = list(), ...)
# S3 method for keras.engine.training.Model
mlflow_save_model(model, path, model_spec = list(), conda_env = NULL, ...)
mlflow_save_model(model, path, model_spec = list(), ...)

Arguments

model

The model that will perform a prediction.

path

Destination path where this MLflow compatible model will be saved.

model_spec

MLflow model config this model flavor is being added to.

...

Optional additional arguments.

conda_env

Path to Conda dependencies file.