反转决策树

时间:2019-07-30 20:09:07

标签: r tree rpart

我想知道是什么定义了“黑猫”类,但是到目前为止,决策树只给我提供了不是“黑猫”的类。

我的目标变量是1 =黑猫,0 =所有其他动物。我的数据集非常不平衡(所有其他动物的95%,黑猫的5%)。独立变量都是分类的:颜色(红色,蓝色,黑色,白色,...)和动物(猫,狗,...)。

我希望得到以下结果: 节点1:颜色将树分成2个。黑色的动物在树的左边,其他的都在右边。 节点2:动物将分支分成2个。猫在左边,其他的都在右边,所以我的左边是黑猫。

我的问题是,在每个节点上,我的树都显示了不是黑猫的类别。例如: 节点1:红色/蓝色/白色/黄色/粉红色/ ... 节点2:鲸鱼/狐狸/狗/老鼠/...

在rpart参数中,是否有一种方法可以强制树在每个节点上显示什么是黑猫而不是黑猫?

fit <- rpart(BlackCat ~ Animal + Color, method="class", data=mydata)
rpart.plot(fit)

0 个答案:

没有答案