當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。