- 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
- experiment_name_formatter
- databricks_permissions
- conda_env
- timeout
- timestamp_column_name
Model name formatter.
Experiment name formatter.
Users with read/write permissions.
Conda environment.
Timeout for get requests.
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.
morpheus.controllers.mlflow_model_writer_controller.MLFlowModelWriterController
Previous
morpheus.controllers.mlflow_model_writer_controller
Next
morpheus.controllers.monitor_controller