“ OneHotEncoder”对象没有属性“ get_feature_names”

Question

我正在尝试使用scikit learning的OneHotEncoder对象的get_feature_names函数提取功能，但是这使我出错 “'OneHotEncoder'对象没有属性'get_feature_names'”。

下面是代码段

# Creating the object instance for label encoder
encoder = OneHotEncoder(sparse=False)
onehot_encoded = encoder.fit_transform(df[data_column_category])
onehot_encoded_frame = pd.DataFrame(onehot_encoded,columns = encoder.get_feature_names(data_column_category))

Answer 1

该功能是最近引入的，因此您可能只需要更新sklearn版本。

您可以执行以下操作：

pip install -U scikit-learn