首页 > 解决方案 > rpart 可以同时处理数字数据和分类数据吗?

问题描述

我正在尝试使用分类数据和数字数据在 R 中训练一个模型来预测客户是否购买了东西,当我绘制树来查看拆分时,它完全忽略了性别。

如下所示,我将性别变量编码为 1 和 2。男性和女性之间的比例大致相等。我没有扩展任何功能。

head(df1)
  Gender Age EstimatedSalary Purchased
1      2  19              19         0
2      2  35              20         0
3      1  26              43         0
4      1  27              57         0
5      2  19              76         0
6      2  27              58         0

我可以提供这个显示决策树的链接。

性别对于这个预测根本不重要,还是我错过了其他东西?

标签: rdecision-tree

解决方案


推荐阅读