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


R parsnip rand_forest 隨機森林


rand_forest() 定義了一個模型,該模型創建大量決策樹,每個決策樹彼此獨立。最終的預測使用各個樹的所有預測並將它們組合起來。該函數可以擬合分類、回歸和審查回歸模型。

擬合該模型的方法有多種,通過設置模型引擎來選擇估計方法。下麵列出了該模型的引擎特定頁麵。

1 默認引擎。 ² 需要一個防風草擴展包來進行審查回歸、分類和回歸。

有關如何操作的更多信息防風草用於建模的是https://www.tidymodels.org/.

用法

rand_forest(
  mode = "unknown",
  engine = "ranger",
  mtry = NULL,
  trees = NULL,
  min_n = NULL
)

參數

mode

預測結果模式的單個字符串。此模型的可能值為"unknown"、"regression" 或"classification"。

engine

指定用於擬合的計算引擎的單個字符串。

mtry

創建樹模型時在每次分割時隨機采樣的預測變量數量的整數。

trees

集合中包含的樹數的整數。

min_n

節點中進一步拆分節點所需的最小數據點數量的整數。

細節

此函數僅定義正在擬合的模型類型。一旦指定了引擎,也就定義了擬合模型的方法。有關設置引擎的更多信息,包括如何設置引擎參數,請參閱set_engine()

fit() 函數與數據一起使用之前,模型不會經過訓練或擬合。

此函數中除 modeengine 之外的每個參數都被捕獲為 quosures 。要以編程方式傳遞值,請使用injection operator,如下所示:

value <- 1
rand_forest(argument = !!value)

參考

https://www.tidymodels.org, Tidy Modeling with R, searchable table of parsnip models

例子

show_engines("rand_forest")
#> # A tibble: 6 × 2
#>   engine       mode          
#>   <chr>        <chr>         
#> 1 ranger       classification
#> 2 ranger       regression    
#> 3 randomForest classification
#> 4 randomForest regression    
#> 5 spark        classification
#> 6 spark        regression    

rand_forest(mode = "classification", trees = 2000)
#> Random Forest Model Specification (classification)
#> 
#> Main Arguments:
#>   trees = 2000
#> 
#> Computational engine: ranger 
#> 
源代碼:R/rand_forest.R

相關用法


注:本文由純淨天空篩選整理自Max Kuhn等大神的英文原創作品 Random forest。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。