如何使用DecisionTreeRegressor（）进行分类分割？

Question

我已经使用python的DecisionTreeRegressor()根据连续的预测器对数据进行分段，并且效果很好。 在本项目中，我被要求使用分类数据作为预测器。 预测 - 行业领域，响应 - 收入。 使用DecisionTreeRegressor()时会抛出错误

  

“无法将字符串更改为浮动：行业域名”。

您能否建议是否有办法解决此问题？

Answer 1

对于这种决策树，您需要使用DecisionTreeClassifier（）。看起来DecisionTreeRegressor仅适用于数字预测数据。 DecisionTreeClassifier（）仅适用于类预测变量数据。我真的想要一个能做到这两点的人，但它似乎不可能。