0df2cf25a625d05b7590efdbbe335498b93d694d,finetune/target_encoders.py,OrdinalRegressionEncoder,fit,#OrdinalRegressionEncoder#Any#,77

Before Change

        
    def fit(self, x):
        x = np.array(x)
        rank = len(x.shape)
        if rank == 1:
            x = np.expand_dims(x, 1)
        self.fit_transform(x)
        return self

After Change

        return np.array(x, dtype=np.int32).reshape(-1, 1)

    def fit(self, x):
        super().fit(self._force_2d(x))
        self.num_outputs = len(self.categories_[0]) - 1
        return self

    def transform(self, x):

Instances