Importancia de variables en árboles

Los árboles (o árboles de inferencia condicional) valen fundamentalmente para hacerse una idea de cómo y en qué grado opera una variable en un modelo controlando por el efecto del resto. Su valor reside fundamentalmente en la interpretabilidad.

No obstante lo cual, no es infrecuente construir árboles muy grandes. Y el tamaño dificulta censar qué variables y en qué manera aparecen. Por eso me vi obligado recientemente a crear un pequeño prototipo para extraer el peso de las variables de un árbol.

(Nótese que uso cursivas para peso: lo he definido yo arbitrariamente y depende esencialmente de la altura a la que se encuentre la variable, i.e., es mayor cuanto más arriba aparece y obviamente del número de veces que aparece la variable, sumándose los pesos de cada aparición en el conjunto).

He aquí el código incrustado en un ejemplo:

library(party)
library(ggplot2)
library(plyr)

irisct <- ctree(Species ~ .,data = iris)
irisct
plot(irisct)

ctree.varimp <- function(x,n = 0){
  if(is.null(x$psplit$variableName))
    return(NULL)

  res <- list(node = x$psplit$variableName, depth = n)

  c(list(res), ctree.varimp(x$left, n+1), ctree.varimp(x$right, n+1))
}

res <- ctree.varimp(irisct@tree)
res <- do.call(rbind, lapply(res, as.data.frame))
res$depth <- max(res$depth) + 1 - res$depth

res <- ddply(res, .(node), summarize, importancia = sum(depth))
res$node <- reorder( res$node, res$importancia, max )

ggplot(res, aes(x = node, weight = importancia)) + geom_bar() +
  coord_flip() + ggtitle("Importancia de variables")

El resultado es: