min_grid.tabnet: Determine the minimum set of model fits

In tabnet: Fit 'TabNet' Models for Classification and Regression

Determine the minimum set of model fits

Description

min_grid() determines exactly what models should be fit in order to evaluate the entire set of tuning parameter combinations. This is for internal use only and the API may change in the near future.

Usage

## S3 method for class 'tabnet'
min_grid(x, grid, ...)

Arguments

x	A model specification.
grid	A tibble with tuning parameter combinations.
...	Not currently used.

Details

fit_max_value() can be used in other packages to implement a min_grid() method.

Value

A tibble with the minimum tuning parameters to fit and an additional list column with the parameter combinations used for prediction.

Examples

library(dials)
library(tune)
library(parsnip)

tabnet_spec <- tabnet(decision_width = tune(), attention_width = tune()) %>%
  set_mode("regression") %>%
  set_engine("torch")

tabnet_grid <-
  tabnet_spec %>%
  extract_parameter_set_dials() %>%
  grid_regular(levels = 3)

min_grid(tabnet_spec, tabnet_grid)

tabnet documentation built on April 17, 2025, 1:07 a.m.