mlflow_save_model: Save Model for MLflow

In mlflow: Interface to 'MLflow'

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 class 'crate'
mlflow_save_model(model, path, model_spec = list(), ...)

mlflow_save_model(model, path, model_spec = list(), ...)

## S3 method for class 'H2OModel'
mlflow_save_model(model, path, model_spec = list(), conda_env = NULL, ...)

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

## S3 method for class 'xgb.Booster'
mlflow_save_model(model, path, model_spec = list(), conda_env = NULL, ...)

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.

mlflow documentation built on Nov. 1, 2024, 1:07 a.m.