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.
Selon cet article , un tracé partiel est réalisable avec gbm.
Merci d'avance pour votre aide.
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
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 = "")