搜索内容

有一个问题?

如果您有任何疑问,可以在下面询问或输入您要寻找的!

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

技术聚客 未结 1 1838
OXOYO
OXOYO 2021-03-08 10:49

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

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

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

1条回答
  • 2021-03-08 11:11

    你这个情况是属于多标签的one-hot encoding,在sklearn里面叫MultiLabelBinarizer,同时喜欢的情况把两列都置为1。

    0 讨论(0)
提交回复