search

bethatkinson / rpart

Recursive Partitioning and Regression Trees
43 stars 23 forks source link

what does "split 1" mean in parms for rpart.object? #56

Open A-Pai opened 4 months ago

A-Pai commented 4 months ago

library(rpart) progstat <- factor(stagec$pgstat, levels = 0:1, labels = c("No", "Prog")) cfit <- rpart(progstat ~ age + eet + g2 + grade + gleason + ploidy,data = stagec, method = 'class') cfit$parms

image

Does "split 1" means "gini"?

A-Pai commented 4 months ago

Yeah, just like I thought:

fit1 <- rpart(Kyphosis ~ Age + Number + Start, data = kyphosis, parms = list(split = "gini")) fit1 $parms fit2 <- rpart(Kyphosis ~ Age + Number + Start, data = kyphosis, parms = list(split = "information")) fit2 $parms

image