Skip to content

OneHotEncoder.get_feature_names doesn't work with integer column names #16593

Open
@amueller

Description

@amueller
import pandas as pd
import numpy as np
from sklearn.preprocessing import OneHotEncoder
X = pd.DataFrame({1: np.random.randint(0, 10, size=400)})
OneHotEncoder().fit(X).get_feature_names([1])
TypeError: unsupported operand type(s) for +: 'int' and 'str'

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions