nnf_multilabel_one_hot: Convert class_id tensor to binary one-hot tensor
In tabnet: Fit 'TabNet' Models for Classification and Regression
nnf_multilabel_one_hot R Documentation
Convert class_id tensor to binary one-hot tensor
Description
Transforms a tensor of class indices (one column per hierarchy level)
into a binary tensor where each column corresponds to a class.
Usage
nnf_multilabel_one_hot(y, outcomes, device = "cpu")
Arguments
y
A torch_tensor of shape (batch_size, n_levels) containing
1-based class indices.
outcomes
A tibble with factor columns (as from hardhat::mold()$outcomes).
device
Torch device.
Value
A torch_tensor of shape (batch_size, n_classes) with binary values.
tabnet documentation built on June 12, 2026, 5:06 p.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.
| nnf_multilabel_one_hot | R Documentation |
Convert class_id tensor to binary one-hot tensor
Description
Transforms a tensor of class indices (one column per hierarchy level) into a binary tensor where each column corresponds to a class.
Usage
nnf_multilabel_one_hot(y, outcomes, device = "cpu")
Arguments
y |
A |
outcomes |
A tibble with factor columns (as from |
device |
Torch device. |
Value
A torch_tensor of shape (batch_size, n_classes) with binary values.
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.