NVIDIA 文档中心 NVIDIA Morpheus NVIDIA Morpheus (24.10.01) MLflow 模型写入模块

MLflow 模型写入模块

此模块将训练好的模型上传到 MLflow 服务器。

可配置参数

参数	类型	描述	示例值	默认值
`conda_env`	字符串	模型的 Conda 环境	`"path/to/conda_env.yml"`	`[必需]`
`databricks_permissions`	字典	模型的权限	请参考下方	`无`
`experiment_name_formatter`	字符串	实验名称的格式化器	`"experiment_name_{timestamp}"`	`[必需]`
`model_name_formatter`	字符串	模型名称的格式化器	`"model_name_{timestamp}"`	`[必需]`
`timestamp_column_name`	字符串	时间戳列的名称	`"timestamp"`	`timestamp`

databricks_permissions

键	类型	描述	示例值	默认值
`read`	数组	具有读取权限的用户列表	`["read_user1", "read_user2"]`	`-`
`write`	数组	具有写入权限的用户列表	`["write_user1", "write_user2"]`	`-`

示例 JSON 配置

复制
已复制！

            
            {
  "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"]
  }
}

上一篇 FromControlMessage 模块

下一篇批量数据载荷模块