Rpart作为虚拟矩阵

时间:2017-07-12 22:58:57

标签: r rpart

考虑以下rpart

library(rpart)
fit <- rpart(Kyphosis ~ Age + Number, data = kyphosis)

我的问题是:如何改变以下内容:

fit$splits

进入虚拟矩阵。例如。我明白了:

> fit$split
       count ncat  improve index adj
Number    81   -1 2.866795   5.5   0
Age       81   -1 2.250212  39.5   0

所以虚拟矩阵将有一列:

ifelse(Age > 39.5, 1, 0)

同样适用于Number

ifelse(Number > 5.5, 1, 0)

0 个答案:

没有答案