1a19396d3daea14a189c4335e5df9f4072a764cc,dirty_cat/similarity_encoder.py,SimilarityEncoder,fit,#SimilarityEncoder#,143

Before Change

                                                 dtype=object))
                
        if self.categories == "auto":
            self.categories_ = [le.classes_ for le in self._label_encoders_]

        return self

    def transform(self, X):

After Change

        for i in range(n_features):
            Xi = X[:, i]
            if self.categories == "auto":
                self.categories_.append(np.unique(Xi))
            elif self.categories == "most_frequent":
                self.categories_.append(self.get_most_frequent(Xi))
            else:

Instances