Scikit-优化如何将收敛图保存到文件

Question

我使用Scikit-Optimize库实现了贝叶斯优化，并希望在运行后保存一些图。我尝试使用以下代码：

res = gp_minimize(objective, space, n_calls = 10, acq_func="EI", verbose = True)
print(res.x)
print(res.fun)
ax1 = plot_objective(res)
ax1.savefig("objective.png")

但我得到错误：

AttributeError: 'numpy.ndarray' object has no attribute 'savefig'

我已经在不同的地方寻找过如何做到这一点的例子，但我遇到的是人们做的事情：

_ = plot_objective(forest_res)

并显示结果图但这并没有显示他们如何保存它。

Answer 1

使用matplotlib.pyplot保存当前图表（可以导入为plt）完成

plt.savefig("filename.png")

如果要保存由返回numpy数组轴的函数创建的特定图形，

axes = func(...)
axes.flatten()[0].figure.savefig("filename.png")