Teilabhängigkeitsdiagramme und Gradientenverstärkung (GBM-Paket)

9

Ist es möglich, ein partielles Abhängigkeitsdiagramm zu zeichnen, um die Klassenwahrscheinlichkeit anzuzeigen und die Auswirkungen eines Prädiktors für ein GBM- Modell abzuschätzen ? Ähnliches wie partialPlotaus dem randomForestPaket.

Geben Sie hier die Bildbeschreibung ein

Nach diesem Artikel ist ein Teilplot mit gbm machbar.

Vielen Dank im Voraus für Ihre Hilfe.

Panos Kal.
quelle

Antworten:

12

Ja, das ist wirklich einfach. Auschecken ?plot.gbm. Die Verwendung sieht folgendermaßen aus:

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

und sieht aus wie Geben Sie hier die Bildbeschreibung ein

i.varsteuert den Index der zu zeichnenden Variablen. Sie können auch Zwei-Wege-Diagramme (oder mehr) erstellen

 plot(gbm.model, i.var = c(1,2), lwd = 2, col = "blue", main = "")
wcampbell
quelle
1
Vielen Dank, ich schätze Ihre Bemühungen und Ihre Zeit hier und in Ihrem Blog sehr.
Panos Kal.