python - 如何以编程方式可视化路德维希库模型学习曲线？

Question

我正在使用 Uber 的ludwig库来训练神经网络模型。我想以编程方式（通过 python 代码，使用文档的示例）可视化我的模型的学习曲线，以使用以下代码分析它们的性能：

import ludwig
ludwig.visualize.learning_curves(
  [train_stats],
  TARGET,
  model_names=None,
  output_directory=None,
  file_format='pdf'
)

但是，运行代码时出现以下错误：

AttributeError: module 'ludwig' has no attribute 'visualize'

我知道我可以通过命令行生成可视化。但是，我需要以编程方式进行。

score 0 · Accepted Answer

那样做。

from ludwig import visualize
visualize.learning_curves(
  [train_stats],
  TARGET,
  model_names=None,
  output_directory=None,
  file_format='pdf'
)

score 0 · Accepted Answer

我能够在不更改代码的情况下运行相同的代码。我刚刚更新了 ludwig 的版本，但我不确定这是否能解决我的问题。我一直在使用这段代码并且它有效：

import ludwig
ludwig.visualize.learning_curves(
  [train_stats],
  TARGET,
  model_names=None,
  output_directory=None,
  file_format='pdf'
)

2 回答 2