我们很高兴宣布 MLflow 1.12.0 已发布!
除了错误修复和文档改进之外,MLflow 1.12.0 还包括几项主要功能和改进,特别是 MLflow 的 PyTorch 集成和自动日志记录方面的诸多改进。
PyTorch
mlflow.pytorch.log_model
、mlflow.pytorch.load_model
现在支持记录/加载 TorchScript 模型 (#3557, @shrinath-suresh)mlflow.pytorch.log_model
支持传递requirements_file
和extra_files
参数,以便在模型之外记录额外的工件 (#3436, @shrinath-suresh)
自动日志记录
- 添加通用
mlflow.autolog
,该 API 为所有支持的集成启用自动日志记录 (#3561, #3590, @andrewnitu) - 添加
mlflow.pytorch.autolog
API,用于自动记录 Pytorch Lightning 训练中的指标、参数和模型 (#3601, @shrinath-suresh, #3636, @karthik-77)。此 API 也通过mlflow.autolog
启用。 - Scikit-learn、XGBoost 和 LightGBM 的自动日志记录现在支持记录模型签名和输入示例 (#3386, #3403, #3449, @andrewnitu)
mlflow.sklearn.autolog
现在支持记录指标(例如准确度)和图表(例如混淆矩阵热力图)(#3423, #3327, @willzhan-db, @harupy)
有关更改的完整列表,请参阅发布日志,并查看 mlflow.org 上的最新文档。