如何保存pycaret图？

Question

我正在尝试保存pycaret图，但只得到了一个空文件：

plt.figure(figsize = (18,9))
plot_model(pycaret_xgboost, 'auc')
plt.savefig('ROC_xboost.svg')

Answer 1

当您尝试将图形另存为svg文件时，可能会出现此问题。StackOverflow上的答案是Here，它可能会对您有所帮助

Answer 2

将PyCaret更新到2.0

将pycaret更新为2.0 (pip install pycaret==2.0)。在plot_model()函数中增加了一个新的参数save。将save设置为True时，将根据绘图类型将png / html文件保存在当前工作目录中。每个绘图的名称都是预先定义的。

Answer 3

在PyCaret 2.0中，您可以在plot_model内部传递save参数。

例如：

plot_model(model, plot = 'AUC', save=True)

这会将AUC.png保存在您当前的工作目录中。