R/models.R
predict_contributions.H2OModel.RdDefault implemntation return H2OFrame shape (#rows, #features + 1) - there is a feature contribution column for each input feature, the last column is the model bias (same value for each row). The sum of the feature contributions and the bias term is equal to the raw prediction of the model. Raw prediction of tree-based model is the sum of the predictions of the individual trees before the inverse link function is applied to get the actual prediction. For Gaussian distribution the sum of the contributions is equal to the model prediction.
predict_contributions.H2OModel( object, newdata, output_format = c("original", "compact"), top_n = 0, bottom_n = 0, compare_abs = FALSE, ... ) h2o.predict_contributions( object, newdata, output_format = c("original", "compact"), top_n = 0, bottom_n = 0, compare_abs = FALSE, ... )
| object | a fitted H2OModel object for which prediction is desired |
|---|---|
| newdata | An H2OFrame object in which to look for variables with which to predict. |
| output_format | Specify how to output feature contributions in XGBoost - XGBoost by default outputs contributions for 1-hot encoded features, specifying a compact output format will produce a per-feature contribution. Defaults to original. |
| top_n | Return only #top_n highest contributions + bias If top_n<0 then sort all SHAP values in descending order If top_n<0 && bottom_n<0 then sort all SHAP values in descending order |
| bottom_n | Return only #bottom_n lowest contributions + bias If top_n and bottom_n are defined together then return array of #top_n + #bottom_n + bias If bottom_n<0 then sort all SHAP values in ascending order If top_n<0 && bottom_n<0 then sort all SHAP values in descending order |
| compare_abs | True to compare absolute values of contributions |
| ... | additional arguments to pass on. |
Returns an H2OFrame contain feature contributions for each input row.
Note: Multinomial classification models are currently not supported.
h2o.gbm and h2o.randomForest for model
generation in h2o.
# NOT RUN { library(h2o) h2o.init() prostate_path <- system.file("extdata", "prostate.csv", package = "h2o") prostate <- h2o.uploadFile(path = prostate_path) prostate_gbm <- h2o.gbm(3:9, "AGE", prostate) h2o.predict(prostate_gbm, prostate) # Compute SHAP h2o.predict_contributions(prostate_gbm, prostate) # Compute SHAP and pick the top two highest h2o.predict_contributions(prostate_gbm, prostate, top_n=2) # Compute SHAP and pick the top two lowest h2o.predict_contributions(prostate_gbm, prostate, bottom_n=2) # Compute SHAP and pick the top two highest regardless of the sign h2o.predict_contributions(prostate_gbm, prostate, top_n=2, compare_abs=TRUE) # Compute SHAP and pick the top two lowest regardless of the sign h2o.predict_contributions(prostate_gbm, prostate, bottom_n=2, compare_abs=TRUE) # Compute SHAP values and show them all in descending order h2o.predict_contributions(prostate_gbm, prostate, top_n=-1) # Compute SHAP and pick the top two highest and top two lowest h2o.predict_contributions(prostate_gbm, prostate, top_n=2, bottom_n=2) # }