morpheus.controllers.mlflow_model_writer_controller.MLFlowModelWriterController

class MLFlowModelWriterController(model_name_formatter, experiment_name_formatter, databricks_permissions, conda_env, timeout, timestamp_column_name)[source]

Bases: object

Controller class for writing machine learning models to MLflow with optional permissions and configurations.

Parameters
model_name_formatterstr

Model name formatter.

experiment_name_formatterstr

Experiment name formatter.

databricks_permissionsdict

Users with read/write permissions.

conda_envdict

Conda environment.

timeout

Timeout for get requests.

timestamp_column_name

Timestamp column name to be used from the dataframe.

Attributes
databricks_permissions

experiment_name_formatter

model_name_formatter

Methods

on_data(message) Stores incoming models into MLflow.
user_id_to_experiment(user_id) Converts a user ID to an experiment name
user_id_to_model(user_id) Converts a user ID to an model name
on_data(message)[source]

Stores incoming models into MLflow.

Parameters
messageMultiAEMessage

The incoming message containing the model and related metadata.

Returns
MultiAEMessage

The processed message.

user_id_to_experiment(user_id)[source]

Converts a user ID to an experiment name

Parameters
user_idstr

The user ID.

Returns
str

The generated experiment name.

user_id_to_model(user_id)[source]

Converts a user ID to an model name

Parameters
user_idstr

The user ID.

Returns
str

The generated model name.

Previous morpheus.controllers.mlflow_model_writer_controller
Next morpheus.controllers.monitor_controller
© Copyright 2023, NVIDIA. Last updated on Feb 2, 2024.