treeshap.model_unified fails with data.table input

In treeshap.model_unified function, this line does not behave as expected if x is a data.table :

x <- x[,colnames(x) %in% unified_model$feature_names]

x=data.table(a=1:3,b=2:4)
x[,colnames(x)%in%colnames(x)]
[1] TRUE TRUE
x%>%as.data.frame%>%.[,colnames(x)%in%colnames(x)]
  a b
1 1 2
2 2 3
3 3 4

I suggest to convert x to a data.frame when x is a data.table

You seem to ask how to select columns in {data.table}.

I can help you with this:

library(data.table)

x <- data.table(a = 1:3, b = 2:4)
x[, colnames(x) %in% colnames(x), with = FALSE]

Since the question seems completely unrelated to {treeshap}, I am closing the issue.

Navigation Menu