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


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