如何以编程方式可视化路德维希图书馆模型学习曲线？

Question

我正在使用 Uber 的 ludwig library to train neural network models. I want to programatically (via python code, using the documentation's example) 可视化我的模型的学习曲线，以使用以下代码分析它们的性能：

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'

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

Answer 1

那样做。

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

Answer 2

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

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

如何以编程方式可视化路德维希图书馆模型学习曲线？

How to visualize ludwig library models learning curves programatically?

python

machine-learning

ludwig