如何解决能否获得 R 的 TidyModel 框架中随机森林模型的袋外误差?
如果直接使用 ranger 函数,可以从得到的 ranger 类对象中获取袋外误差。
如果相反,人们通过设置配方、模型规范/引擎、调整参数等方式继续,我们如何提取相同的错误? Tidymodels 方法似乎无法保留这些数据。
解决方法
如果你想访问欧洲防风草对象内部的游侠对象,它是$fit
:
library(tidymodels)
data("ad_data",package = "modeldata")
rf_spec <-
rand_forest() %>%
set_engine("ranger",oob.error = TRUE) %>%
set_mode("classification")
rf_fit <- rf_spec %>%
fit(Class ~ .,data = ad_data)
rf_fit
#> parsnip model object
#>
#> Fit time: 158ms
#> Ranger result
#>
#> Call:
#> ranger::ranger(x = maybe_data_frame(x),y = y,oob.error = ~TRUE,num.threads = 1,verbose = FALSE,seed = sample.int(10^5,1),probability = TRUE)
#>
#> Type: Probability estimation
#> Number of trees: 500
#> Sample size: 333
#> Number of independent variables: 130
#> Mtry: 11
#> Target node size: 10
#> Variable importance mode: none
#> Splitrule: gini
#> OOB prediction error (Brier s.): 0.1340793
class(rf_fit)
#> [1] "_ranger" "model_fit"
class(rf_fit$fit)
#> [1] "ranger"
rf_fit$fit$prediction.error
#> [1] 0.1340793
由 reprex package (v1.0.0) 于 2021 年 3 月 11 日创建
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。