Package ai.djl.basicmodelzoo.tabular
Contains classes for the tabular section (
Application.Tabular
) of the BasicModelZoo
.-
Class Summary Class Description TabNet TabNet
contains a generic implementation of TabNet adapted from https://towardsdatascience.com/implementing-tabnet-in-pytorch-fc977c383279 (Original author Samrat Thapa)TabNet.AttentionTransformer 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.TabNet.Builder The Builder to construct aTabNet
object.TabNet.DecisionStep DecisionStep is just combining featureTransformer and attentionTransformer together.