[Rd] rpart package, text function, and round of class counts
Achim Zeileis
Achim.Zeileis at uibk.ac.at
Mon Mar 5 07:35:15 CET 2012
On Sun, 4 Mar 2012, yindalon wrote:
> I run the following code:
>
> library(rpart)
> data(kyphosis)
> fit <- rpart(Kyphosis ~ ., data=kyphosis)
> plot(fit)
> text(fit, use.n=TRUE)
>
> The text labels represent the count of each class at the leaf node.
> Unfortunately, the numbers are rounded and in scientific notation rather
> than the exact number of examples sorted by that node in each class.
You probably have a getOption("digits") of 4 or lower. text.rpart uses
getOption("digits") - 3 as the default which then means only 1 significant
digit and hence it rounds and uses scientific notation. Using
text(fit, use.n = TRUE, digits = 3)
should do the trick. Maybe adding setting xpd = TRUE in addition helps in
avoiding clipping of some labels.
Also, I would recommend to use
library("partykit")
plot(as.party(fit))
for visualization which uses a display like for the ctree() function (also
mentioned on the web page you quote below).
> The plot is supposed to look like
> http://www.statmethods.net/advstats/images/ctree.png as per
> http://www.statmethods.net/advstats/cart.html.
>
> I'm running 2.14.1 on a mac.
>
> Can anyone verify or point out if I am doing something obviously wrong for
> displaying the counts rounded and in scientific notation rather than the
> true counts in each class at each node?
> Thanks.
>
> --
> View this message in context: http://r.789695.n4.nabble.com/rpart-package-text-function-and-round-of-class-counts-tp4444576p4444576.html
> Sent from the R devel mailing list archive at Nabble.com.
>
> ______________________________________________
> R-devel at r-project.org mailing list
> https://stat.ethz.ch/mailman/listinfo/r-devel
>
More information about the R-devel
mailing list