butcher (version 0.4.0)

axe-tabnet_fit: Axing a tabnet_fit.

Description

Axing a tabnet_fit.

Remove fitted values.

Usage

# S3 method for `_tabnet_fit`
axe_fitted(x, verbose = FALSE, ...)

Value

Axed tabnet_fit object.

Arguments

x

A model object.

verbose

Print information each time an axe method is executed. Notes how much memory is released and what functions are disabled. Default is FALSE.

...

Any additional arguments related to axing.

Examples

if (FALSE) { # rlang::is_installed("tabnet")

# Load libraries
suppressWarnings(suppressMessages(library(parsnip)))
suppressWarnings(suppressMessages(library(rsample)))
suppressWarnings(suppressMessages(library(tabnet)))

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


if (interactive() & torch::torch_is_installed()) {
  torch::torch_manual_seed(1)

  # Create model and fit
  mtcar_fit <- tabnet::tabnet() |>
  set_mode("regression") |>
  set_engine("torch") |>
  fit(mpg ~ ., data = car_train)

  out <- butcher(mtcar_fit, verbose = TRUE)
}
}