我可以用R Cubist结果制作一个图表吗?

时间:2013-07-10 01:53:10

标签: r diagram regression decision-tree

我刚开始使用R包Cubist,它创建了这样的输出:

  

Cubist [发布2.07 GPL版] Tue Jul 09 19:46:48 2013

     

目标属性`结果'

     

从undefined.data

中读取260个案例(9个属性)      

型号:

     

规则1:[26例,平均值0.3,范围0至8,估计值为0.3]

if
    B4 <= 54.96766
    B7 > 39.66716
then
    outcome = 0
     

规则2:[48例,平均值0.3,范围0至8,估计值为0.6]

if
    B1 > 56.99043
    B5 > 74.11118
    B6 > 155.996
then
    outcome = 9.1 - 0.17 B5 + 0.25 B7 + 0.15 NDVI

我可以使用预测来使用模型,但我想制作树的图形。我没有看到可以查看手册,但我想知道是否有人知道如何做到这一点。

2 个答案:

答案 0 :(得分:1)

您可以使用dotplot.cubist来获取模型条件和系数的直观视图。这是一个例子:

library(mlbench)
library(Cubist)
library(gridExtra)
data(BostonHousing)
mod1 <- cubist(x = BostonHousing[, -14], y = BostonHousing$medv)
summary(mod1)
p1 <- dotplot(mod1, what = "splits",main='Conditions')
p2 <- dotplot(mod1, what = "coefs",main='Coefs')
grid.arrange(p1,p2)

enter image description here

答案 1 :(得分:0)

rpartpartykit都是用于绘制树样式图的有用R包。

编辑:请参阅这里的partykit示例here