search

pbiecek / ceterisParibus

Ceteris Paribus Plots (What-If plots) for explanations of a single observation
https://pbiecek.github.io/ceterisParibus/
41 stars 13 forks source link

error for xgboost #4

Closed agosiewska closed 6 years ago

agosiewska commented 6 years ago 
library("breakDown")
head(wine)

library("xgboost")

model_martix_train <- model.matrix(quality ~ . - 1, wine)
data_train <- xgb.DMatrix(model_martix_train, label = wine$quality)
param <- list(max_depth = 2, eta = 1, silent = 1, nthread = 2, objective = "reg:linear")

wine_xgb_model <- xgb.train(param, data_train, nrounds = 50)

library("DALEX")

explainer_xgb <- explain(wine_xgb_model, 
                         data = model_martix_train, 
                         y = wine$quality, 
                         label = "xgboost")

library("ceterisParibus")
nobs <- model_martix_train[1, , drop = FALSE]
cp_xgb  <- ceteris_paribus(explainer_xgb, observation = nobs)

returns

Error in data[, vname] : subscript out of bounds

agosiewska commented 6 years ago

Thank you for fixing it.