如何保存pycaret图？

Question

我正在尝试保存一个pycaret生成的图表，但保存下来的文件却是空白的：

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

相关问题：

暂无相关问题

Answer 1

这里附上一个例子，展示了如何使用yellowbrick从pycaret模型中生成向量图（pdf、svg、eps）。这个例子已经在googlecolab和pycaret 2.3.4上测试过了。

#---- data
from pycaret.datasets import get_data
dataset = get_data('diamond')

#---- model pycaret
from pycaret.regression import *
exp_reg101 = setup(data = dataset, target = 'Price', session_id=123, 
                   remove_multicollinearity = True, multicollinearity_threshold = 0.95) 

lightgbm = create_model('lightgbm')

#---- plot
from yellowbrick.regressor import ResidualsPlot

X_train, X_test, y_train, y_test = get_config('X_train'), get_config('X_test'), get_config('y_train'), get_config('y_test')

visualizer = ResidualsPlot(lightgbm) # regression model
visualizer.fit(X_train, y_train) # Fit the training data to the visualizer
visualizer.score(X_test, y_test) # Evaluate the model on the test data
visualizer.poof(outpath="ResidualsPlot.pdf") # VECTOR .pdf .eps .svg, RASTER .png .jpg .tif
#visualizer.poof()  # Finalize and render the figure

#---- file extensions supported
import matplotlib.pyplot as plt
plt.figure().canvas.get_supported_filetypes()

Answer 2

@maria_g 提供了一个正确的参考。在这个回答中，提问者只使用了 pyplot 来绘制和保存图表。

Answer 3

更新 PyCaret 到 2.0

要把 pycaret 更新到 2.0 版本，你可以使用这个命令：pip install pycaret==2.0。在 plot_model() 这个函数里，新增了一个参数 save。当你把 save 设置为 True 时，程序会根据你绘制的图的类型，把一个 png 或 html 文件保存在当前的工作目录里。每个图的名称都是预先定义好的。

Answer 4

在PyCaret 2.0中，你可以在plot_model这个函数里添加一个叫save的参数。

举个例子：

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

这样做会把一个名为AUC.png的文件保存在你当前的工作目录里。

Answer 5

当你尝试把图形保存为svg文件时，可能会遇到这个问题。这里有一个StackOverflow上的回答，可能对你有帮助。

如何保存pycaret图？

相关问题：

6 个回答

更新 PyCaret 到 2.0

撰写回答