Tracés de dépendance partielle et renforcement du gradient (package GBM)

9

Est-il possible de tracer un graphique de dépendance partielle pour afficher la probabilité de classe et estimer les effets d'un prédicteur pour un modèle GBM ? Quelque chose de similaire à celui partialPlotdu randomForestpackage.

entrez la description de l'image ici

Selon cet article , un tracé partiel est réalisable avec gbm.

Merci d'avance pour votre aide.

r  gbm 
— Panos Kal.
source

Réponses:

12

Oui, c'est vraiment facile. Découvrez ?plot.gbm. L'utilisation va comme ceci:

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

et ressemble entrez la description de l'image ici

i.varcontrôle l'index de la variable à tracer. Vous pouvez également faire des tracés à deux voies (ou plus) en utilisant

 plot(gbm.model, i.var = c(1,2), lwd = 2, col = "blue", main = "")
— wcampbell
source
1
merci, j'apprécie vraiment vos efforts et votre temps ici et dans votre blog.
— Panos Kal.
En utilisant notre site, vous reconnaissez avoir lu et compris notre politique liée aux cookies et notre politique de confidentialité.
Licensed under cc by-sa 3.0 with attribution required.