Нужна помощь по объектам в PyCaret

Я изучаю PyCaret. Это фантастическая программа. У меня следующие вопросы:

1) После запуска lr = create_model('lr', fold = 5), Получается хорошая таблица с точностью и т. Д.

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

Как мне найти данные в этом классе sklearn.multiclass.OneVsRestClassifier. Есть ли способ конвертировать данные в pd dataframe для экспорта?

2)plot_model(lr) создает красивый график AUC. fig=plot_model(lr)также создает такой же красивый график. Но

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

поэтому я не могу сохранить рисунок и экспортировать его. Я могу добавитьsave=True в код, но это сохранит рисунок как auc.png, что нормально, но я хотел бы сохранить рисунки с другими именами.

1 ответ

Если вы спрашиваете, как сохранить сетку очков, которая печатается при использовании create_model, вы можете использовать pullфункциональность. Например:

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

В настоящее время нет возможности изменить имя графика при сохранении. Вы можете использоватьsave параметр внутри plot_model чтобы сохранить его, но он будет использовать имя по умолчанию PyCaret For, например AUC.png для участка AUC.

Другие вопросы по тегам