当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


R baguette predict.bagger 来自 bagged 模型的预测


predict() 函数计算集合中每个模型的预测,并为 new_data 中的每个样本返回单个聚合值。

用法

# S3 method for bagger
predict(object, new_data, type = NULL, ...)

参数

object

bagger() 生成的对象。

new_data

预测变量的 DataFrame 。如果最初使用配方或公式,则应在此处传递原始数据而不是预处理版本。

type

预测类型的单个字符值。对于回归模型,type = 'numeric' 有效,而 'class''prob' 对于分类模型有效。

...

目前未使用。

例子

data(airquality)

set.seed(7687)
mars_bag <- bagger(Ozone ~ ., data = airquality, base_model = "MARS", times = 5)
predict(mars_bag, new_data = airquality[, -1])
#> # A tibble: 153 × 1
#>    .pred
#>    <dbl>
#>  1  38.6
#>  2  19.8
#>  3  15.8
#>  4  18.5
#>  5 NaN  
#>  6 NaN  
#>  7  25.2
#>  8  14.4
#>  9  12.4
#> 10  31.0
#> # ℹ 143 more rows
源代码:R/predict.R

相关用法


注:本文由纯净天空筛选整理自Max Kuhn等大神的英文原创作品 Predictions from a bagged model。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。