Package ai.djl.basicmodelzoo.tabular

package ai.djl.basicmodelzoo.tabular

Contains classes for the tabular section (Application.Tabular) of the BasicModelZoo.

Related Packages

Package	Description
ai.djl.basicmodelzoo	Contains the built-in BasicModelZoo.
ai.djl.basicmodelzoo.basic	Contains the basic built-in models.
ai.djl.basicmodelzoo.nlp	Contains classes for the natural language processing section (Application.NLP) of the BasicModelZoo.

Classes

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 a TabNet object.
TabNet.DecisionStep	DecisionStep is just combining featureTransformer and attentionTransformer together.