gbm::interact.gbm vs. dismo::gbm.interactions

前端 未结 1 1032
孤独总比滥情好
孤独总比滥情好 2021-01-01 18:16

Background

The reference manual for the gbm package states the interact.gbm function computes Friedman\'s H-statistic to a

相关标签:
1条回答
  • 2021-01-01 18:50

    To summarize, the difference between the two approaches boils down to how the "partial dependence function" of the two predictors is estimated.

    The dismo package is based on code originally given in Elith et al., 2008 and you can find the original source in the supplementary material. The paper very briefly describes the procedure. Basically the model predictions are obtained over a grid of two predictors, setting all other predictors at their means. The model predictions are then regressed onto the grid. The mean squared errors of this model are then multiplied by 1000. This statistic indicates departures of the model predictions from a linear combination of the predictors, indicating a possible interaction.

    From the dismo package, we can also obtain the relevant source code for gbm.interactions. The interaction test boils down to the following commands (copied directly from source):

    interaction.test.model <- lm(prediction ~ as.factor(pred.frame[,1]) + as.factor(pred.frame[,2]))
    
    interaction.flag <- round(mean(resid(interaction.test.model)^2) * 1000,2)
    

    pred.frame contains a grid of the two predictors in question, and prediction is the prediction from the original gbm fitted model where all but two predictors under consideration are set at their means.

    This is different than Friedman's H statistic (Friedman & Popescue, 2005), which is estimated via formula (44) for any pair of predictors. This is essentially the departure from additivity for any two predictors averaging over the values of the other variables, NOT setting the other variables at their means. It is expressed as a percent of the total variance of the partial dependence function of the two variables (or model implied predictions) so will always be between 0-1.

    0 讨论(0)
提交回复
热议问题