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


R butcher axe-xgb.Booster 砍掉 xgb.Booster。


xgb.Booster 對象是從創建的xgboost包,它提供了梯度提升決策樹的高效且可擴展的實現。考慮到後處理函數對模型對象的依賴,例如xgb.Booster.complete,在列出的第一類中,butcher_xgb.Booster不附加類。

用法

# S3 method for xgb.Booster
axe_call(x, verbose = FALSE, ...)

# S3 method for xgb.Booster
axe_env(x, verbose = FALSE, ...)

參數

x

一個模型對象。

verbose

每次執行 ax 方法時打印信息。記錄釋放了多少內存以及禁用了哪些函數。默認為 FALSE

...

與砍伐相關的任何其他參數。

取消了 xgb.Booster 對象。

例子

library(xgboost)
#> 
#> Attaching package: ‘xgboost’
#> The following object is masked from ‘package:dplyr’:
#> 
#>     slice
library(parsnip)

data(agaricus.train)
bst <- xgboost(data = agaricus.train$data,
               label = agaricus.train$label,
               eta = 1,
               nthread = 2,
               nrounds = 2,
               eval_metric = "logloss",
               objective = "binary:logistic",
               verbose = 0)

out <- butcher(bst, verbose = TRUE)
#> ✔ Memory released: 31.12 kB
#> ✖ Disabled: `print()`, `summary()`, and `xgb.Booster.complete()`
#> ✖ Could not add <butchered> class

# Another xgboost model
fit <- boost_tree(mode = "classification", trees = 20) %>%
  set_engine("xgboost", eval_metric = "mlogloss") %>%
  fit(Species ~ ., data = iris)

out <- butcher(fit, verbose = TRUE)
#> ✖ The butchered object is 1.13 kB larger than the original. Do not butcher.
源代碼:R/xgb.R

相關用法


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