我的Python自定义转换器不工作

Question

我用Python编写了这个定制的转换器。目的是将其应用到流水线类中进行序列数据预处理步骤。我的数据集有9个数字，第10列是分类的。在

from sklearn.base import BaseEstimator, TransformerMixin

class DataFrameSelector(BaseEstimator, TransformerMixin):
    def _init_(self, attribute_names):
       self.attribute_names = attribute_names
    def fit(self, X, y=None):
        return self
    def transform(self, X):
        return X[self.attribute_names].values

在我试图运行这段代码时定义了这个类之后，我得到了下面列出的错误

仅供参考。。。。datasets_num是一个只包含数字列/属性的数据帧。在

^{pr2}$

错误：

Traceback (most recent call last):

  File "<ipython-input-34-f509d02ccc6e>", line 7, in <module>
     ('selector', DataFrameSelector(num_attributes)),

 TypeError: object() takes no parameters

Answer 1

这里：

class DataFrameSelector(BaseEstimator, TransformerMixin):
    def _init_(self, attribute_names):

您需要双下划线：

^{pr2}$