构建模型,可以生成所有变量重要性图。但是,当我在连续变量上使用partialPlot
时,例如purchase_value
:
partialPlot(rf, fraud_data_train, purchase_value, which.class = 1)
错误是
Error in is.finite(x): default method not implemented for type 'list'
对于分类变量(browser
),错误是
partialPlot(rf, fraud_data_train, browser, which.class = 1)
Error in FUN(X[[i]], ...) :
only defined on a data frame with all numeric variables
数据可用here,代码如下:
rf = randomForest(y = fraud_data_train$class_factor,
x = fraud_data_train[,-predictors_notinclude],
ntree = 30, mtry = 4, keep.forest = TRUE,
importance = TRUE, proximity = TRUE)
partialPlot(rf, fraud_data_train, purchase_value, which.class =1)
更新
更新2
答案 0 :(得分:11)
如果您的数据是tibble
,而不是data.frame
,则会遇到此问题。似乎tibble
(也许data_frame
,但无法测试)与partialPlot
函数不兼容。
一个简单的解决方案是执行partialPlot(rf, as.data.frame(fraud_data_train), purchase_value, which.class = 1)
。