bgreenwell/pdp

Is it possible to do pdp for an xgboost count::poisson? I get the following error:

Martafenix opened this issue · 3 comments

Hi @Martafenix, yes you certainly can. Here’s an example: https://github.com/bgreenwell/pdp/blob/master/slowtests/slowtests-mtcars.R. Not sure why you’re getting an error though without seeing a reproducible example.

My matrix is built like this, maybe that's why:

dt_train <- train_data %>%
select(-all_of(vars_to_remove)) %>%
as.matrix() %>%
xgboost::xgb.DMatrix(label = target_train, weight = weight_train)

Are you able to post a short reproducible example @Martafenix? If so, I'd be happy to provide the fix.