0

I'm begginer in SciKitLearn, and I'm making my first transformer, but when I run it I'm getting error Transformer doesn't take any arguments

class DataFrameSelector(BaseEstimator, TransformerMixin):

    def init(self, attribute_names):
        self.attribute_names = attribute_names

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

    def transform(self,Х):
        return np.array(X[self.attribute_names])

1 Answers1

1

You should write __init__ instead of init.

RO-ito
  • 111
  • 5