This module uploads trained models to the MLflow server.
Parameter |
Type |
Description |
Example Value |
Default Value |
---|---|---|---|---|
|
string |
Conda environment for the model |
“path/to/conda_env.yml” |
|
|
dictionary |
Permissions for the model |
See Below |
|
|
string |
Formatter for the experiment name |
“experiment_name_{timestamp}” |
|
|
string |
Formatter for the model name |
“model_name_{timestamp}” |
|
|
string |
Name of the timestamp column |
“timestamp” |
|
Key |
Type |
Description |
Example Value |
Default Value |
---|---|---|---|---|
|
array |
List of users with read permissions |
[“read_user1”, “read_user2”] |
|
|
array |
List of users with write permissions |
[“write_user1”, “write_user2”] |
|
{
"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"]
}
}