Question

我在R中创建了一个决策树。当我绘制它时，我只有3个节点（1个根和2个终端）。我用来创建决策树的公式是

 >FertilityTree <- rpart(Output~ Age + Surgery + RDrugs + SpermCount +      Smoker, data = FertilityTree, method = "class")

制作的图形是： http://rpubs.com/BonitaWilliams/fertilitydecisiontree

你能帮我看一个显示更多节点的图形吗？或者告诉我为什么我有这么少的节点？

Answer 1

可以提供参数以向分裂算法提供可以增加（或减少）节点数量的进一步信息。如果您有一个特定的结果，预计错误会更加昂贵，那么您需要为其分配不同的loss。没有示例数据就没有经过测试的示例代码，但这种代码的形式是“丢失”。规范将是这样的：

FertilityTree <- rpart(Output~ Age + Surgery + RDrugs + SpermCount +
                          Smoker, 
                       data = FertilityTree, method = "class", 
                       parms=list(loss=matrix(c(0,1,2,0),2) )

我不能保证它会在SO（或宇宙中以先到者为准）结束时可用，但此时你可以找到discussion and worked example here.损失矩阵需要具有正轴外和零轴上元素。还有其他参数可以调整，包括调整先前的＆＃39;并将拆分标准更改为＆＃34;信息＆＃34;。有关这些选项的要求，请参阅parms的{{1}}部分。还可以将拆分标准更改为用户提供的版本。 Terry Therneau posted this in 2011 on Rhelp.

还可以使用?rpart，这是一个以该名称作为参数提供的列表。您可以尝试rpart.control作为允许进一步拆分的简单第一步。更改复杂性参数rpart.control=list(minsplit=10)也可能会产生一些影响。这是帮助页面中的cp部分::

Usage

R中决策树中的节点 - 需要更多节点

1 个答案: