get_constr_output: Apply hierarchy constraints via max-pooling over descendants...
In tabnet: Fit 'TabNet' Models for Classification and Regression
get_constr_output R Documentation
Apply hierarchy constraints via max-pooling over descendants (MCM)
Description
Given neural network outputs x and ancestor matrix R, enforces that
if a class is predicted positive, all its ancestors must also be positive.
Implements: final_out[i] = max{x[j] : R[i,j] = 1}
Usage
get_constr_output(x, R)
Arguments
x
A torch_tensor of shape (batch_size, n_classes).
R
A torch_tensor of shape (1, n_classes, n_classes) where
R[1, i, j] = 1 iff class i is a descendant of class j.
Value
A torch_tensor of shape (batch_size, n_classes) with constrained outputs.
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.
| get_constr_output | R Documentation |
Apply hierarchy constraints via max-pooling over descendants (MCM)
Description
Given neural network outputs x and ancestor matrix R, enforces that
if a class is predicted positive, all its ancestors must also be positive.
Implements: final_out[i] = max{x[j] : R[i,j] = 1}
Usage
get_constr_output(x, R)
Arguments
x |
A |
R |
A |
Value
A torch_tensor of shape (batch_size, n_classes) with constrained outputs.
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.