在PyCaret中的对象上需要帮助

问题描述

我正在学习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，这是可以的，但是我想用不同的名称保存图形。

解决方法

如果您询问如何存储使用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图。