Package ai.djl.basicmodelzoo.tabular
Contains classes for the tabular section (
Application.Tabular) of the BasicModelZoo.-
Class Summary Class Description TabNet TabNetcontains 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 aTabNetobject.TabNet.DecisionStep DecisionStep is just combining featureTransformer and attentionTransformer together.