试图解释虹膜随机森林模型时的DALEX错误

Question

我想用RandomForest模型的Dalex创建解释器，但是当我尝试的时候，我得到了这个错误：

Error: $ operator is invalid for atomic vectors

有人知道如何解决这个问题吗？

下面是我的代码：

library(DALEX)
library(ranger) 
model <- ranger(Species ~ .,data = iris,probability =T,classification = T)
explanation <- explain(model,
                      data=iris,
                      y = iris$Species,
                      model_info = "classification",
                      label="Random Forest")

谢谢!

编辑:我已经尝试过这段代码

library(DALEX)
library(ranger)
model <- ranger(Species ~ .,iris)
model_info <- list(package = "ranger", ver = "0.12.1", type = "classification")

explanation <- explain(model,
                       data=iris,
                       y = iris$Species,
                      
                       label="Random Forest")


plot(model_profile(explanation))

但这并不打印变量的影响。

Answer 1

您可以删除或使用model_info，如下所示：

# needs to be a list. Not all items are needed. You could just use type.
model_info <- list(package = "ranger", ver = "0.12.1", type = "classification")

explanation <- explain(model,
                      data=iris,
                      y = iris$Species,
                      model_info = model_info,
                      label="Random Forest")

不过，当我这样做时，我会收到一条警告消息，因为y(物种)是一个因素。如果从物种中删除因子或不使用model_info，则不会收到此警告消息。