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


R butcher axe-rpart 砍掉一个零件。


rpart 对象是从创建的部分包,用于分类、回归和生存树的递归分区。

用法

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

# S3 method for rpart
axe_ctrl(x, verbose = FALSE, ...)

# S3 method for rpart
axe_data(x, verbose = FALSE, ...)

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

参数

x

一个模型对象。

verbose

每次执行 ax 方法时打印信息。记录释放了多少内存以及禁用了哪些函数。默认为 FALSE

...

与砍伐相关的任何其他参数。

删除 rpart 对象。

例子

# Load libraries
library(parsnip)
library(rsample)
library(rpart)

# Load data
set.seed(1234)
split <- initial_split(mtcars, prop = 9/10)
car_train <- training(split)

# Create model and fit
rpart_fit <- decision_tree(mode = "regression") %>%
  set_engine("rpart") %>%
  fit(mpg ~ ., data = car_train, minsplit = 5, cp = 0.1)

out <- butcher(rpart_fit, verbose = TRUE)
#> ✔ Memory released: 17.62 kB

# Another rpart object
wrapped_rpart <- function() {
  some_junk_in_environment <- runif(1e6)
  fit <- rpart(Kyphosis ~ Age + Number + Start,
               data = kyphosis,
               x = TRUE, y = TRUE)
  return(fit)
}

# Remove junk
cleaned_rpart <- axe_env(wrapped_rpart(), verbose = TRUE)
#> ✔ Memory released: 8.06 MB

# Check size
lobstr::obj_size(cleaned_rpart)
#> 53.42 kB
源代码:R/rpart.R

相关用法


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