Is it possible to do pdp for an xgboost count::poisson? I get the following error:
Martafenix opened this issue · 3 comments
Martafenix commented
bgreenwell commented
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.
Martafenix commented
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)
bgreenwell commented
Are you able to post a short reproducible example @Martafenix? If so, I'd be happy to provide the fix.