sklearn OneHotEncoder broken- ValueError：无法将字符串转换为float

Posted 2021-05-03

我从sklearn OneHotEncoder纪录页面中获取了这个例子：

from sklearn.preprocessing import OneHotEncoder
enc = OneHotEncoder(handle_unknown='ignore')
X = [['Male', 1], ['Female', 3], ['Female', 2]]
enc.fit(X)
enc.categories_

enc.transform([['Female', 1], ['Male', 4]]).toarray()

enc.inverse_transform([[0, 1, 1, 0, 0], [0, 0, 0, 1, 0]])

enc.get_feature_names()

我明白了：

ValueError：无法将字符串转换为float：'Male'。

当我用数字替换“男性”和“女性”：X = [['5', 1], ['4', 3], ['4', 2]]

我明白了：

AttributeError：'OneHotEncoder'对象没有属性'categories_'

我的sklearn版本是0.19.1有人可以重现这个吗？

答案

正如Vivek Kumar所说，0.19.1太老了。升级到0.20.1版解决了这个问题