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

问题描述

问 题

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

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

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

解决方案

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

这篇关于python - sklearn的决策树模型能不能处理多类特征呢？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！