All functions

autoplot.tabnet_explain()

Plot tabnet_explain mask importance heatmap

autoplot.tabnet_fit() autoplot.tabnet_pretrain()

Plot tabnet_fit model loss along epochs

check_compliant_node()

Check that Node object names are compliant

nn_prune_head.tabnet_fit() nn_prune_head.tabnet_pretrain()

Prune top layer(s) of a tabnet network

node_to_df()

Turn a Node object into predictor and outcome.

tabnet()

Parsnip compatible tabnet model

tabnet_config()

Configuration for TabNet models

tabnet_explain()

Interpretation metrics from a TabNet model

tabnet_fit()

Tabnet model

tabnet_nn()

TabNet Model Architecture

decision_width() attention_width() num_steps() feature_reusage() num_independent() num_shared() momentum() mask_type()

Parameters for the tabnet model

tabnet_pretrain()

Tabnet model