b5034279b48ae96ffdd4714f96e0f62b0f4807fc,category_encoders/ordinal.py,OrdinalEncoder,ordinal_encoding,#,239

Before Change

                categories_dict = {x: i + 1 for i, x in enumerate(categories)}

                mapping_out.append({"col": col, "mapping": list(categories_dict.items()), "data_type": X[col].dtype}, )

        return X, mapping_out

After Change

                    categories = [x for x in pd.unique(X[col].values) if x is not None]

                index = []
                values = []

                for i in range(len(categories)):
                    index.append(categories[i])
                    values.append(i + 1)

                mapping = pd.Series(data=values, index=index)

                mapping_out.append({"col": col, "mapping": mapping, "data_type": X[col].dtype}, )

Instances