时间戳

首页 > 解决方案 > 如何在 sklearn 中修复这个自定义转换器？

python - 如何在 sklearn 中修复这个自定义转换器？

问题描述

我编写了这个简单的自定义转换器，用 0 填充特定列中的 na。当我在我的数据集上 fit_transform 时，它不会在指定列中填充 nas。我没有在我的代码中看到问题。

class CustomImputer(BaseEstimator, TransformerMixin): 
    def fit(self, X, y=None):
        return self
    def transform(self, X, y=None):
        for col in ('PavedDrive', 'GarageQual', 'GarageFinish', 'FireplaceQu', 'KitchenQual', 'CentralAir', 'HeatingQC', 'BsmtExposure', 'BsmtCond', 'BsmtQual', 'ExterCond', 'ExterQual', 'Street'):
            X[col].fillna(0)
        return X

我希望返回的数据框是指定列用 0 填充 nas 的数据框，但是，我得到了一个具有相同空值的数据框。

标签： pythonmachine-learningscikit-learnsklearn-pandas

解决方案

您从未将fillna操作分配给新变量。使用

X[col].fillna(0)

不会发生在原地。而是使用：

X.loc[:,col] = X[col].fillna(0)

推荐阅读