Нужна помощь по объектам в 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.