考虑以下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)