跳到内容

第三方集成

使用集成

Ludwig 提供了可扩展的接口,用于与第三方系统集成。要激活某个集成,只需在命令行中插入其标志即可。每个集成可能有特定的要求和用例。

可用集成

MLFlow

--mlflow - 将训练指标、超参数优化参数、输出工件和训练好的模型记录到 MLflow。设置环境变量 MLFLOW_TRACKING_URI 可将结果记录到远程跟踪服务器。

Weights & Biases

--wandb - 将训练指标、配置参数、环境详情和训练好的模型记录到 Weights & Biases。更多详情请参阅 W&B 快速入门

Comet ML

--comet - 将训练指标、环境详情、测试结果、可视化等内容记录到 Comet.ML。需要免费账户。更多详情请参阅 Comet 的文档 将 Ludwig 与 Comet 配合使用

Aim

--aim - 提供包含配置、元数据、超参数、损失和终端日志的完整实验跟踪。要查看端到端的 Aim-Ludwig 训练和跟踪示例,请参考我们的 demo。有关 Aim 的更多详情,请参阅文档

添加更多集成

有关集成贡献的更多信息,请参阅开发者指南