有人试过使用iml软件包中的Shapley函数和ranger回归模型吗?

有人试过使用iml软件包中的Shapley函数和ranger回归模型吗?,r,machine-learning,user-defined-functions,r-ranger,R,Machine Learning,User Defined Functions,R Ranger,我正在尝试使用iml软件包来构建Shapely绘图。根据和文章中给出的文档,如果所使用的模型不是来自支持的软件包(如caret),则必须定义自定义预测函数。当我试图用一个用于回归的ranger模型进行曲线拟合时,我得到了一个错误 我正在执行的代码: featuresData错误消息表示您的自定义pred功能有问题。我尝试了你对predFunction的定义,结果还是同样的错误。 过去,当我为iml使用ranger时,我用一种简单的方式定义了predFunction,如下所示 predFuncti

我正在尝试使用iml软件包来构建Shapely绘图。根据和文章中给出的文档,如果所使用的模型不是来自支持的软件包(如caret),则必须定义自定义预测函数。当我试图用一个用于回归的ranger模型进行曲线拟合时,我得到了一个错误

我正在执行的代码:
featuresData错误消息表示您的自定义
pred功能有问题。我尝试了你对
predFunction
的定义,结果还是同样的错误。 过去,当我为
iml
使用
ranger
时,我用一种简单的方式定义了
predFunction
,如下所示

predFunction <- function(object, newdata) predict(object, data = newdata)$predictions

predFunction我不知道如何直接将Ranger与Shapley一起使用,我使用了注释中提到的解决方法:使用插入符号的Ranger直通车功能。现在我可以画出shapley图了。我仍然想知道是否有人可以直接使用ranger和shapley
predictorRf <- Predictor$new(model = rfModel, data = featuresData, y = responseData, predict.fun = predFunction)
predFunction <- function(object, newdata) predict(object, data = newdata)$predictions