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	Refer 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

{
  "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"]
  }
}