Gráficos de dependência parcial e aumento de gradiente (pacote GBM)

9

É possível plotar um gráfico de dependência parcial para exibir a probabilidade da classe e estimar os efeitos de um preditor para um modelo GBM ? Algo semelhante ao partialPlotdo randomForestpacote.

insira a descrição da imagem aqui

De acordo com este artigo , uma plotagem parcial é possível com o gbm.

Agradeço antecipadamente por sua ajuda.

Panos Kal.
fonte

Respostas:

12

Sim, é realmente fácil. Confira ?plot.gbm. O uso é assim:

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

e parece insira a descrição da imagem aqui

i.varcontrola o índice da variável a ser plotada. Você também pode fazer gráficos de duas (ou mais) maneiras usando

 plot(gbm.model, i.var = c(1,2), lwd = 2, col = "blue", main = "")
wcampbell
fonte
1
obrigado, eu realmente aprecio seus esforços e tempo aqui e em seu blog.
Panos Kal.