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.

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 = "")

1
gracias, realmente aprecio sus esfuerzos y tiempo aquí y en su blog.
Panos Kal.
Al usar nuestro sitio, usted reconoce que ha leído y comprende nuestra Política de Cookies y Política de Privacidad.
Licensed under cc by-sa 3.0 with attribution required.