Index

A B G M N P S
All Classes and Interfaces|All Packages|Constant Field Values|Serialized Form

A

ai.djl.mxnet.zoo - package ai.djl.mxnet.zoo

Contains the built-in MxModelZoo.

ai.djl.mxnet.zoo.nlp.embedding - package ai.djl.mxnet.zoo.nlp.embedding

Contains classes for the natural language processing section (Application.NLP) of the MxModelZoo.

ai.djl.mxnet.zoo.nlp.embedding.utils - package ai.djl.mxnet.zoo.nlp.embedding.utils

Contains utilities for the natural language processing section (Application.NLP) of the MxModelZoo.

ai.djl.mxnet.zoo.nlp.qa - package ai.djl.mxnet.zoo.nlp.qa

Contains classes for the Application.NLP.QUESTION_ANSWER models in the MxModelZoo.

B

build() - Method in class ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslator.Builder

Builds the translator.

builder() - Static method in class ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslator

Creates a builder to build a MxBertQATranslator.

builder(Map<String, ?>) - Static method in class ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslator

Creates a builder to build a MxBertQATranslator.

Builder() - Constructor for class ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslator.Builder

BuildModelZooWordEmbedding - Class in ai.djl.mxnet.zoo.nlp.embedding.utils

A utility to build embeddings into DJL format from Gluon-NLP format.

G

getBatchifier() - Method in class ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslator

getGroupId() - Method in class ai.djl.mxnet.zoo.MxModelZoo

getModelZoo() - Method in class ai.djl.mxnet.zoo.MxZooProvider

getSupportedEngines() - Method in class ai.djl.mxnet.zoo.MxModelZoo

getSupportedTypes() - Method in class ai.djl.mxnet.zoo.nlp.embedding.GloveWordEmbeddingTranslatorFactory

getSupportedTypes() - Method in class ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslatorFactory

GloveWordEmbeddingBlockFactory - Class in ai.djl.mxnet.zoo.nlp.embedding

A BlockFactory class that creates Glove word embedding block.

GloveWordEmbeddingBlockFactory() - Constructor for class ai.djl.mxnet.zoo.nlp.embedding.GloveWordEmbeddingBlockFactory

GloveWordEmbeddingTranslatorFactory - Class in ai.djl.mxnet.zoo.nlp.embedding

A TranslatorFactory that creates a GloveWordEmbeddingTranslatorFactory.GloveWordEmbeddingTranslator instance.

GloveWordEmbeddingTranslatorFactory() - Constructor for class ai.djl.mxnet.zoo.nlp.embedding.GloveWordEmbeddingTranslatorFactory

GROUP_ID - Static variable in class ai.djl.mxnet.zoo.MxModelZoo

M

main(String[]) - Static method in class ai.djl.mxnet.zoo.nlp.embedding.utils.BuildModelZooWordEmbedding

Runs main build embeddings after editing.

MxBertQATranslator - Class in ai.djl.mxnet.zoo.nlp.qa

The translator for MXNet BERT QA model.

MxBertQATranslator.Builder - Class in ai.djl.mxnet.zoo.nlp.qa

The builder for Bert QA translator.

MxBertQATranslatorFactory - Class in ai.djl.mxnet.zoo.nlp.qa

A TranslatorFactory that creates a MxBertQATranslator instance.

MxBertQATranslatorFactory() - Constructor for class ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslatorFactory

MxModelZoo - Class in ai.djl.mxnet.zoo

MxModelZoo is a repository that contains all MXNet models in MxSymbolBlock for DJL.

MxZooProvider - Class in ai.djl.mxnet.zoo

An MXNet model zoo provider implements the ZooProvider interface.

MxZooProvider() - Constructor for class ai.djl.mxnet.zoo.MxZooProvider

N

newBlock(Model, Path, Map<String, ?>) - Method in class ai.djl.mxnet.zoo.nlp.embedding.GloveWordEmbeddingBlockFactory

newInstance(Class<I>, Class<O>, Model, Map<String, ?>) - Method in class ai.djl.mxnet.zoo.nlp.embedding.GloveWordEmbeddingTranslatorFactory

newInstance(Class<I>, Class<O>, Model, Map<String, ?>) - Method in class ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslatorFactory

P

prepare(TranslatorContext) - Method in class ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslator

processInput(TranslatorContext, QAInput) - Method in class ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslator

processOutput(TranslatorContext, NDList) - Method in class ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslator

S

self() - Method in class ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslator.Builder

Returns the builder.

setSeqLength(int) - Method in class ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslator.Builder

Set the max length of the sequence to do the padding.

A B G M N P S
All Classes and Interfaces|All Packages|Constant Field Values|Serialized Form