Question

我想绘制个人条件期望（ICE），并且具有以下代码段：

MSE

我最后收到的图看起来像这样：

在某些ICE在线图上，该图看起来并不像在下图所示那样好：

任何想法为什么会这样？我相信我所拥有的数据至少在价值方面类似于以上文章中显示的数据。

Answer 1

问题在于预测变量会给出类标签而不是类概率。

更改

predictor <- Predictor$new(rf_model, data=X, y=dataset_train$X9)

到

predictor <- Predictor$new(rf_model, data=X, y=dataset_train$X9, type = "prob")

应该修复您的情节。