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:
objectController 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
- messageControlMessage
The incoming message containing the model and related metadata.
- Returns
- ControlMessage
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.