MLflow Model Writer Module

NVIDIA Docs Hub NVIDIA Morpheus NVIDIA Morpheus MLflow Model Writer Module

This module uploads trained models to the MLflow server.

Configurable Parameters

Parameter	Type	Description	Example Value	Default Value
`conda_env`	string	Conda environment for the model	“path/to/conda_env.yml”	`[Required]`
`databricks_permissions`	dictionary	Permissions for the model	See Below	`None`
`experiment_name_formatter`	string	Formatter for the experiment name	“experiment_name_{timestamp}”	`[Required]`
`model_name_formatter`	string	Formatter for the model name	“model_name_{timestamp}”	`[Required]`
`timestamp_column_name`	string	Name of the timestamp column	“timestamp”	`timestamp`

databricks_permissions

Key	Type	Description	Example Value	Default Value
`read`	array	List of users with read permissions	[“read_user1”, “read_user2”]	`-`
`write`	array	List of users with write permissions	[“write_user1”, “write_user2”]	`-`

Example JSON Configuration

Copy
Copied!

            
            {
  "model_name_formatter": "model_name_{timestamp}",
  "experiment_name_formatter": "experiment_name_{timestamp}",
  "conda_env": "path/to/conda_env.yml",
  "timestamp_column_name": "timestamp",
  "databricks_permissions": {
    "read": ["read_user1", "read_user2"],
    "write": ["write_user1", "write_user2"]
  }
}