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

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

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

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

阅读 5.3k
1 个回答

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

撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题