MLflow Keras 3.0 集成

引言

Keras 是一个使用 Python 编写的深度学习 API，运行在机器学习平台 TensorFlow 之上。它最初的开发重点是实现快速实验。

Keras 3.0 使得在 TensorFlow、JAX 和 PyTorch 之上运行 Keras 工作流成为可能。它还允许您将 Keras 组件（如图层、模型或指标）无缝集成到低级别的 TensorFlow、JAX 和 PyTorch 工作流中。

MLflow 为 Keras 3.0 工作流提供内置支持。它提供了一个回调函数，允许您在模型训练期间记录参数和指标。目前尚不支持模型记录。

MLflow + Keras 3.0 5 分钟快速入门

要快速了解如何使用 MLflow + Keras 3.0，请阅读快速入门指南。

Keras 3.0 + MLflow 入门

了解如何利用 MLflow Keras 回调进行实验跟踪，以及如何对其进行自定义。