如何保存 pycaret 图？

Question

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

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

Answer 1

在 PyCaret 2.0 中，您可以save在plot_model.

例如：

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

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