美文网首页
sklearn中OneHotEncoder

sklearn中OneHotEncoder

作者: 阳光夜风 | 来源:发表于2017-07-13 17:35 被阅读89次

OneHotEncoder输入必须是int数组,所以直接传入字符串特征值是不行的,需要先通过LabelEncoder转化成整型特征,再传入OneHotEncoder。

le = LabelEncoder()
df.feature = le.fit_transform(df.feature)

ohe = OneHotEncoder()
trans_feature = ohe.fit_transform(df)
image.png

相关文章

网友评论

      本文标题:sklearn中OneHotEncoder

      本文链接:https://www.haomeiwen.com/subject/bcswhxtx.html