用R中的CHAID进行预测

时间:2017-11-25 15:03:25

标签: r decision-tree predict

我已经使用CHAID构建了一个模型。现在我想用它来预测变量 - 就像我用树()或rpart()做的那样:

MA_CHAID <- chaid(Data$target
              ~ Data$factor1
              + Data$factor2,
              control = chaid_control(minprob = 0.001,
                                      minsplit = 500,minbucket = 200),
              data = Data
               )

Data$MA_predict <- predict(MA_CHAID, Data)

但我收到此错误消息:

  

错误'$&lt; - .data.frame'(' tmp ',MA_predict,value = c(1L,1L,:   替换有124855行,数据有379031

有什么想法吗?

谢谢!

1 个答案:

答案 0 :(得分:2)

构建CHAID树,包括缺少值的预测变量。

该算法排除了所有缺少值的行。这就是为什么模型只用一半的情况建造的原因。

我的下一次尝试将使用&#34;缺少&#34;作为自己的一个类别。