如何在pycaret中保存compare_models()中的模型比较数据框?

Joh*_*Tam 2 pandas pycaret

我想从 pycaret 中的compare_models()保存模型比较数据框。

# load dataset
from pycaret.datasets import get_data
diabetes = get_data('diabetes')

# init setup
from pycaret.classification import *
clf1 = setup(data = diabetes, target = 'Class variable')

# compare models
best = compare_models()
Run Code Online (Sandbox Code Playgroud)

在此输入图像描述

即这个数据框如上所示。

有谁知道这是怎么做到的吗?

Joh*_*Tam 6

解决方案是:

df = pull()
Run Code Online (Sandbox Code Playgroud)

作者:Goosang Yu,来自 pycaret slack 社区。