猿问

sklearn的决策树模型能不能处理多类特征呢?

比如有一个特征,喜欢的运动
小明喜欢足球
小花喜欢足球和篮球
小白喜欢篮球

      喜欢的运动
小明  足球
小花  篮球和足球
小白  篮球

那么这个该怎么存呢?向量还是数组?
在官网没有找到,只找到一个向量,看了一下感觉又不太像,没看太懂
sklearn.feature_extraction.DictVectorizer
求大神指点!

繁星coding
浏览 542回答 1
1回答

慕莱坞森

你这个情况是属于多标签的one-hot encoding,在sklearn里面叫MultiLabelBinarizer,同时喜欢的情况把两列都置为1。
随时随地看视频慕课网APP

相关分类

Python
我要回答