Sklearn的SimpleImputer在管道中时无法检索插补值

问题描述 投票:1回答:1

我尝试用SimpleImputer拟合后输出所有插补值。单独使用SimpleImputer时,我可以从实例的statistics_属性中检索这些。

这很好用:

s = SimpleImputer(strategy='mean')
s.fit(df[['feature_1', 'feature_2']])
print(s.statistics_)

但是,在管道中使用SimpleImputer时,我无法这样做。

这不起作用:

numeric_transformer = Pipeline(steps=[
    ('simple_imputer', SimpleImputer(strategy='mean')),
    ('scaler', StandardScaler())])

categorical_features = ['feature_3']
categorical_transformer = Pipeline(steps=[
    ('simple_imputer', SimpleImputer(strategy='most_frequent')),
    ('one_hot', OneHotEncoder(handle_unknown='ignore'))])

preprocessor = ColumnTransformer(
    transformers=[
        ('num', numeric_transformer, numeric_features),
        ('cat', categorical_transformer, categorical_features)])

clf = Pipeline(steps=[('preprocessor', preprocessor),
                      ('classifier', RandomForestClassifier(n_estimators=100))])

clf.fit(df[numeric_features + categorical_features], df['target'])

print(clf.named_steps['preprocessor'].transformers[0][1].named_steps['simple_imputer'].statistics_)

我收到以下错误:

AttributeError                            Traceback (most recent call last)
<ipython-input-523-7390eac0d9d6> in <module>
     19 clf.fit(df[numeric_features + categorical_features], df['target'])
     20 
---> 21 print(clf.named_steps['preprocessor'].transformers[0][1].named_steps['simple_imputer'].statistics_)

AttributeError: 'SimpleImputer' object has no attribute 'statistics_

[我相信我正在获取拟合的SimpleImputer对象的正确实例。为什么我无法检索其statistics_属性以打印出插补值?

python scikit-learn pipeline sklearn-pandas imputation
1个回答
0
投票

我发现在使用sklearn管道时更容易使用“点”符号,这尤其重要,因为您会获得自动完成功能来帮助您导航管道的结构/属性。 (我认为)它还具有更多可读性的额外好处。

您可以使用以下行访问statistics_SimpleImputer属性:

clf.named_steps.preprocessor.named_transformers_.num.named_steps.simple_imputer.statistics_
© www.soinside.com 2019 - 2024. All rights reserved.