python - PyCaret 中的对象需要帮助

Question

我正在学习 PyCaret。这是一个很棒的节目。我有以下问题：

1）运行后lr = create_model('lr', fold = 5)，我得到了一个准确的好表等。

>>>print(type(lr))
<class 'sklearn.multiclass.OneVsRestClassifier'>

我怎样才能找到这个类中的数据sklearn.multiclass.OneVsRestClassifier。有什么方法可以转换为 pd 数据框以进行导出？

2)plot_model(lr)创建一个漂亮的 AUC 图。fig=plot_model(lr)也创建了同样漂亮的图表。但

>>>print(type(fig))
<class 'NoneType'>

所以我无法保存图形并将其导出。我可以添加save=True到代码中，但这会将图形保存为 auc.png，这没关系，但我想用不同的名称保存图形。

score 1 · Accepted Answer

如果您询问如何存储使用时打印的分数网格create_model，您可以使用pull功能。例如：

lr = create_model('lr')
lr_results = pull()
type(lr_results) # it will be a pandas DataFrame

目前，保存时无法更改绘图名称。您可以save在内部使用参数plot_model来保存它，但它将使用 PyCaret 的默认命名，例如AUC.png用于 AUC 图。

1 回答 1