跳到主要内容

MLflow 2.1.0

·3 分钟阅读
MLflow maintainers
MLflow 维护者
MLflow 维护者

我们很高兴地宣布 MLflow 2.1.0 现已发布！

MLflow 2.1.0 包含多项重大功能和改进。

功能

  • [Recipes] 支持多类别分类 (#7458, @mshtelma)
  • [Recipes] 扩展分类模型的 pyfunc 表示，以输出除标签外的分数 (#7474, @sunishsheth2009)
  • [UI] 向 Runs 页面添加用户 ID 和生命周期阶段快速搜索链接 (#7462, @jaeday)
  • [Tracking] 对 GetMetricHistory API 进行分页 (#7523, #7415, @BenWilson2)
  • [Tracking] 为 Run 名称和开始时间添加与 UI 列名对应的 Runs 搜索别名 (#7492, @apurva-koti)
  • [Tracking] 向 mlflow server 添加 /version 端点，用于查询服务器的 MLflow 版本 (#7273, @joncarter1)
  • [Model Registry] 为 Model Registry 添加 FileStore 支持 (#6605, @serena-ruan)
  • [Model Registry] 引入 mlflow.search_registered_models() 流式 API (#7428, @TSienki)
  • [Model Registry / Java] 向 Java 客户端添加 getRegisteredModel() 方法 (#6602) (#7511, @drod331)
  • [Model Registry / R] 向 R 客户端添加 mlflow_set_model_version_tag() 方法 (#7401, @leeweijie)
  • [Models] 在 MLmodel 规范和 log_model() 方法中引入 metadata 字段 (#7237, @jdonzallaz)
  • [Models] 扩展 Model.load() 以支持从远程位置加载 MLmodel 规范 (#7517, @dbczumar)
  • [Models] 在 Models 的 requirements.txt 和 conda.yaml 文件中固定 MLflow 的主版本 (#7364, @BenWilson2)
  • [Scoring] 扩展 mlflow.pyfunc.spark_udf() 以支持 StructType 结果 (#7527, @WeichenXu123)
  • [Scoring] 使用 mlflow.pyfunc.spark_udf() 扩展 TensorFlow 和 Keras Models 以支持多维输入 (#7531, #7291, @WeichenXu123)
  • [Scoring] 在部署模型到 SageMaker 时，支持指定部署环境变量和标签 (#7433, @jhallard)

错误修复

  • [Recipes] 修复了在模型调优期间阻止使用自定义 early_stop 函数的 bug (#7538, @sunishsheth2009)
  • [Recipes] 修复了在数据摄取期间创建 Spark Session 时使用的逻辑中的 bug (#7307, @WeichenXu123)
  • [Tracking] 使 mlflow.autolog() 生成的指标名称与 mlflow.evaluate() 保持一致 (#7418, @wenfeiy-db)
  • [Tracking] 修复了导致 XGBoost 和 LightGBM 模型记录嵌套、冗余信息的 autologging bug (#7404, @WeichenXu123)
  • [Tracking] 正确将 SQLAlchemy OperationalErrors 归类为可重试的 HTTP 错误 (#7240, @barrywhart)
  • [Artifacts] 在使用 FTP artifact 存储时，正确处理凭据中的特殊字符 (#7479, @HCTsai)
  • [Models] 解决了阻止 MLeap 模型在 Windows 上保存的问题 (#6966, @dbczumar)
  • [Scoring] 修复了在使用 NFS 进行模型评分（使用 mlflow.pyfunc.spark_udf()）时遇到的权限问题 (#7427, @WeichenXu123)

文档更新

  • [Docs] 在 Runs 搜索文档页面添加更多示例 (#7487, @apurva-koti)
  • [Docs] 为社区开发的 Model flavors 添加文档 (#7425, @mmerce)
  • [Docs] 添加了记录和评分 ONNX Models 的示例 (#7398, @Rusteam)
  • [Docs] 修复了模型评分 REST API 示例中，使用 dataframe_split 格式的输入时的一个拼写错误 (#7540, @zhouyangyu)
  • [Docs] 修复了模型评分 REST API 示例中，使用 dataframe_records 格式的输入时的一个拼写错误 (#7361, @dbczumar)

有关更改的完整列表，请参阅 发布变更日志，并访问 mlflow.org 查看最新文档。