morpheus.controllers.mlflow_model_writer_controller.MLFlowModelWriterController

(Latest Version)
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_formatter

Model name formatter.

experiment_name_formatter

Experiment name formatter.

databricks_permissions

Users with read/write permissions.

conda_env

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
message

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_id

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_id

The user ID.

Returns
str

The generated model name.

Previous morpheus.controllers.mlflow_model_writer_controller
Next morpheus.controllers.monitor_controller
© Copyright 2024, NVIDIA. Last updated on Apr 25, 2024.