Gráficos de dependencia parcial y aumento de gradiente (paquete GBM)

9

¿Es posible trazar un diagrama de dependencia parcial para mostrar la probabilidad de clase y estimar los efectos de un predictor para un modelo GBM ? Algo similar al partialPlotdel randomForestpaquete.

ingrese la descripción de la imagen aquí

Según este artículo , una trama parcial es factible con gbm.

Gracias de antemano por tu ayuda.

Panos Kal.
fuente

Respuestas:

12

Si, es realmente fácil. Echa un vistazo ?plot.gbm. El uso es así:

 plot(gbm.model, i.var = 1, lwd = 2, col = "blue", main = "")

y parece ingrese la descripción de la imagen aquí

i.varcontrola el índice de la variable a trazar. También puede hacer dos (o más) diagramas de manera usando

 plot(gbm.model, i.var = c(1,2), lwd = 2, col = "blue", main = "")
wcampbell
fuente
1
gracias, realmente aprecio sus esfuerzos y tiempo aquí y en su blog.
Panos Kal.