SciKit-Learn CustomTransformer:TypeError:“ numpy.ndarray”对象不可调用

时间:2019-10-09 06:42:09

标签: python pandas machine-learning scikit-learn

我需要创建一个自定义转换器以输入到平地机中。

平地机将字典列表传递给我的估算器而不是DataFrame的predict或predict_proba方法。这意味着该模型必须同时使用两种数据类型。因此,我需要提供一个自定义的ColumnSelectTransformer,以代替使用scikit-learn自己的ColumnTransformer。

这是我的自定义转换器代码,旨在在提供的列中删除空值。

simple_cols = ['BEDCERT', 'RESTOT', 'INHOSP', 'CCRC_FACIL', 'SFF', 'CHOW_LAST_12MOS', 'SPRINKLER_STATUS', 'EXP_TOTAL', 'ADJ_TOTAL']

class ColumnSelectTransformer(BaseEstimator, TransformerMixin):
    def __init__(self, columns):
        self.columns = columns

    def fit(self, X, y=None):
        return self

    def transform(self, X):
        if not isinstance(X, pd.DataFrame):
            X = pd.DataFrame(X)
        X.dropna(inplace=True)
        return X[self.columns].values()

simple_features = Pipeline([
    ('cst', ColumnSelectTransformer(simple_cols)),
])

但是,我无法通过以下断言测试

assert data['RESTOT'].isnull().sum() > 0
assert not np.isnan(simple_features.fit_transform(data)).any()

我产生类型错误

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-44-922f08231b1f> in <module>()
      1 assert not data['RESTOT'].isnull().sum() > 0
----> 2 assert not np.isnan(simple_features.fit_transform(data)).any()

/opt/conda/lib/python3.7/site-packages/sklearn/pipeline.py in fit_transform(self, X, y, **fit_params)
    391                 return Xt
    392             if hasattr(last_step, 'fit_transform'):
--> 393                 return last_step.fit_transform(Xt, y, **fit_params)
    394             else:
    395                 return last_step.fit(Xt, y, **fit_params).transform(Xt)

/opt/conda/lib/python3.7/site-packages/sklearn/base.py in fit_transform(self, X, y, **fit_params)
    551         if y is None:
    552             # fit method of arity 1 (unsupervised transformation)
--> 553             return self.fit(X, **fit_params).transform(X)
    554         else:
    555             # fit method of arity 2 (supervised transformation)

<ipython-input-42-e20ea4310864> in transform(self, X)
     12             X = pd.DataFrame(X)
     13         X.dropna(inplace=True)
---> 14         return X[self.columns].values()
     15 
     16 simple_features = Pipeline([

TypeError: 'numpy.ndarray' object is not callable

如果有人要访问,这里是实际数据。

%%bash
mkdir data
wget http://dataincubator-wqu.s3.amazonaws.com/mldata/providers-train.csv -nc -P ./ml-data
wget http://dataincubator-wqu.s3.amazonaws.com/mldata/providers-metadata.csv -nc -P ./ml-data

data = pd.read_csv('./ml-data/providers-train.csv', encoding='latin1')

1 个答案:

答案 0 :(得分:2)

正如日志所指出的,错误出在 "kind": "adsense#report", "totalMatchedRows": "1", "headers": [ { "name": "EARNINGS", "type": "METRIC_CURRENCY", "currency": "EUR" }, { "name": "CLICKS", "type": "METRIC_TALLY" } ], "rows": [ [ "7.58", "17" ] ], "totals": [ "7.58", "17" ], "averages": [ "7.58", "17" ], "startDate": "2019-10-07", "endDate": "2019-10-07" } 中。 X[self.columns].values()是一个numpy数组,因此您不能将其作为方法来调用(在其后加上括号)。您应该尝试values