如何获取在 PyCaret 中创建的模型的列名称

Question

我正在尝试创建一个 PyCaret 模型

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

# init environment
from pycaret.regression import *

r1 = setup(insurance, target = 'charges', session_id = 123,
           normalize = True,
           polynomial_features = True,
           bin_numeric_features= ['age', 'bmi'])

我看到现在已创建 55 列。如何获取 PyCaret 模型的列名称

Answer 1

训练模型

lr = create_model('lr')

#获取模型列名称

lr.feature_names_in_