Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!
Get Started for Free

tabnet_params

Parameters for the tabnet model

Description

Parameters for the tabnet model

Usage

decision_width(range = c(8L, 64L), trans = NULL)

attention_width(range = c(8L, 64L), trans = NULL)

num_steps(range = c(3L, 10L), trans = NULL)

feature_reusage(range = c(1, 2), trans = NULL)

num_independent(range = c(1L, 5L), trans = NULL)

num_shared(range = c(1L, 5L), trans = NULL)

Arguments

range

the default range for the parameter value
trans

wether to apply a transformation to the parameter

These functions are used with tune grid functions to generate candidates.

Value

A dials parameter to be used when tuning TabNet models.

tabnet

Fit 'TabNet' Models for Classification and Regression

v0.1.0
MIT + file LICENSE
Authors
Daniel Falbel [aut, cre], RStudio [cph]
Initial release

We don't support your browser anymore

Please choose more modern alternatives, such as Google Chrome or Mozilla Firefox.