Package ai.djl.basicmodelzoo.tabular
package ai.djl.basicmodelzoo.tabular
Contains classes for the tabular section (
Application.Tabular
) of the BasicModelZoo
.-
ClassDescription
TabNet
contains a generic implementation of TabNet adapted from https://towardsdatascience.com/implementing-tabnet-in-pytorch-fc977c383279 (Original author Samrat Thapa)AttentionTransformer is where the tabNet models learn the relationship between relevant features, and decides which features to pass on to the feature transformer of the current decision step.The Builder to construct aTabNet
object.DecisionStep is just combining featureTransformer and attentionTransformer together.