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


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.。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。