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.

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

1
obrigado, eu realmente aprecio seus esforços e tempo aqui e em seu blog.
Panos Kal.
Ao utilizar nosso site, você reconhece que leu e compreendeu nossa Política de Cookies e nossa Política de Privacidade.
Licensed under cc by-sa 3.0 with attribution required.