nn_prune_head: Prune top layer(s) of a tabnet network
In tabnet: Fit 'TabNet' Models for Classification and Regression
nn_prune_head.tabnet_fit R Documentation
Prune top layer(s) of a tabnet network
Description
Prune head_size last layers of a tabnet network in order to
use the pruned module as a sequential embedding module.
Usage
nn_prune_head.tabnet_fit(x, head_size)
nn_prune_head.tabnet_pretrain(x, head_size)
Arguments
x
nn_network to prune
head_size
number of nn_layers to prune, should be less than 2
Value
a tabnet network with the top nn_layer removed
tabnet documentation built on June 22, 2024, 10:22 a.m.
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
| nn_prune_head.tabnet_fit | R Documentation |
Prune top layer(s) of a tabnet network
Description
Prune head_size last layers of a tabnet network in order to
use the pruned module as a sequential embedding module.
Usage
nn_prune_head.tabnet_fit(x, head_size)
nn_prune_head.tabnet_pretrain(x, head_size)
Arguments
x |
nn_network to prune |
head_size |
number of nn_layers to prune, should be less than 2 |
Value
a tabnet network with the top nn_layer removed
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
Embedding an R snippet on your website
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.