python - 如何保存 pycaret 图？

Question

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

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

标签： pythonmatplotlibmachine-learningpycaret

Answer 1

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

例如：

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

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