我正在使用Uber的ludwig库来训练神经网络模型。我希望通过编程方式(通过Python代码,使用文档中的示例)来可视化我的模型的学习曲线,以分析它们的性能,使用以下代码:
import ludwigludwig.visualize.learning_curves( [train_stats], TARGET, model_names=None, output_directory=None, file_format='pdf')
然而,运行代码时我得到了以下错误:
AttributeError: module 'ludwig' has no attribute 'visualize'
我知道可以通过命令行生成可视化图表。然而,我需要以编程方式来完成这个任务。
回答:
我能够运行相同的代码而没有做任何更改。我只是更新了ludwig的版本,但我并不确定这是否解决了我的问题。我继续使用这段代码,并且它工作得很好:
import ludwigludwig.visualize.learning_curves( [train_stats], TARGET, model_names=None, output_directory=None, file_format='pdf')