Index

A B C D E F G H L M N O P R S T U V
All Classes and Interfaces|All Packages|Constant Field Values

A

ai.djl.fasttext - package ai.djl.fasttext

Contains the main fastText implementation of the DJL API.

ai.djl.fasttext.jni - package ai.djl.fasttext.jni

Contains classes to interface with the underlying fastText.

ai.djl.fasttext.zoo - package ai.djl.fasttext.zoo

Contains the built-in FtModelZoo.

ai.djl.fasttext.zoo.nlp.textclassification - package ai.djl.fasttext.zoo.nlp.textclassification

Contains classes for the Application.NLP.TEXT_CLASSIFICATION models in the FtModelZoo.

ai.djl.fasttext.zoo.nlp.word_embedding - package ai.djl.fasttext.zoo.nlp.word_embedding

Contains classes for the Application.NLP.WORD_EMBEDDING models in the FtModelZoo.

B

build() - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Builds a new FtTrainingConfig.

builder() - Static method in class ai.djl.fasttext.FtTrainingConfig

Creates a builder to build a FtTrainingConfig.

C

CBOW - Enum constant in enum class ai.djl.fasttext.FtTrainingMode

checkModel(String) - Method in class ai.djl.fasttext.jni.FtWrapper

classify(String) - Method in class ai.djl.fasttext.zoo.nlp.textclassification.FtTextClassification

Returns the classifications of the input text.

classify(String, int) - Method in class ai.djl.fasttext.zoo.nlp.textclassification.FtTextClassification

Returns top K classifications of the input text.

close() - Method in class ai.djl.fasttext.FtAbstractBlock

close() - Method in class ai.djl.fasttext.FtModel

close() - Method in class ai.djl.fasttext.jni.FtWrapper

D

DEFAULT_LABEL_PREFIX - Static variable in class ai.djl.fasttext.zoo.nlp.textclassification.FtTextClassification

describeInput() - Method in class ai.djl.fasttext.FtModel

describeOutput() - Method in class ai.djl.fasttext.FtModel

E

embedWord(NDArray) - Method in class ai.djl.fasttext.zoo.nlp.word_embedding.FtWord2VecWordEmbedding

embedWord(NDManager, long) - Method in class ai.djl.fasttext.zoo.nlp.word_embedding.FtWord2VecWordEmbedding

embedWord(String) - Method in class ai.djl.fasttext.FtAbstractBlock

Embeds a word using fasttext.

F

fit(FtTrainingConfig, RawDataset<Path>) - Static method in class ai.djl.fasttext.zoo.nlp.textclassification.FtTextClassification

Trains the fastText model.

forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.fasttext.zoo.nlp.textclassification.FtTextClassification

forwardInternal(ParameterStore, NDList, boolean, PairList<String, Object>) - Method in class ai.djl.fasttext.zoo.nlp.word_embedding.FtWordEmbeddingBlock

fta - Variable in class ai.djl.fasttext.FtAbstractBlock

FtAbstractBlock - Class in ai.djl.fasttext

A parent class containing shared behavior for SymbolBlocks based on fasttext models.

FtAbstractBlock(FtWrapper) - Constructor for class ai.djl.fasttext.FtAbstractBlock

Constructs a FtAbstractBlock.

FtModel - Class in ai.djl.fasttext

FtModel is the fastText implementation of Model.

FtModel(String) - Constructor for class ai.djl.fasttext.FtModel

Constructs a new Model.

FtModelZoo - Class in ai.djl.fasttext.zoo

FtModelZoo is a repository that contains all fastText models for DJL.

FtTextClassification - Class in ai.djl.fasttext.zoo.nlp.textclassification

A FtAbstractBlock for Application.NLP.TEXT_CLASSIFICATION.

FtTextClassification(FtWrapper, String) - Constructor for class ai.djl.fasttext.zoo.nlp.textclassification.FtTextClassification

Constructs a FtTextClassification.

FtTrainingConfig - Class in ai.djl.fasttext

An interface that is responsible for holding the configuration required by fastText training.

FtTrainingConfig.Builder - Class in ai.djl.fasttext

A builder to construct a FtTrainingConfig.

FtTrainingConfig.FtLoss - Enum Class in ai.djl.fasttext

Loss functions that fastText supports.

FtTrainingMode - Enum Class in ai.djl.fasttext

Training modes that fastText supported.

FtWord2VecWordEmbedding - Class in ai.djl.fasttext.zoo.nlp.word_embedding

An implementation of WordEmbedding for FastText word embeddings.

FtWord2VecWordEmbedding(FtAbstractBlock, Vocabulary) - Constructor for class ai.djl.fasttext.zoo.nlp.word_embedding.FtWord2VecWordEmbedding

Constructs a FtWord2VecWordEmbedding.

FtWord2VecWordEmbedding(Model, Vocabulary) - Constructor for class ai.djl.fasttext.zoo.nlp.word_embedding.FtWord2VecWordEmbedding

Constructs a FtWord2VecWordEmbedding.

FtWordEmbeddingBlock - Class in ai.djl.fasttext.zoo.nlp.word_embedding

A FtAbstractBlock for Application.NLP.WORD_EMBEDDING.

FtWordEmbeddingBlock(FtWrapper) - Constructor for class ai.djl.fasttext.zoo.nlp.word_embedding.FtWordEmbeddingBlock

Constructs a FtWordEmbeddingBlock.

FtWrapper - Class in ai.djl.fasttext.jni

A class containing utilities to interact with the fastText JNI layer.

FtZooProvider - Class in ai.djl.fasttext.zoo

An fastText model zoo provider implements the ZooProvider interface.

FtZooProvider() - Constructor for class ai.djl.fasttext.zoo.FtZooProvider

G

getArtifact(String) - Method in class ai.djl.fasttext.FtModel

getArtifact(String, Function<InputStream, T>) - Method in class ai.djl.fasttext.FtModel

getArtifactAsStream(String) - Method in class ai.djl.fasttext.FtModel

getArtifactNames() - Method in class ai.djl.fasttext.FtModel

getBlock() - Method in class ai.djl.fasttext.FtModel

getBucket() - Method in class ai.djl.fasttext.FtTrainingConfig

Returns number of buckets.

getContextWindow() - Method in class ai.djl.fasttext.FtTrainingConfig

Returns size of the context window.

getDataType() - Method in class ai.djl.fasttext.FtModel

getDevices() - Method in class ai.djl.fasttext.FtTrainingConfig

getDirectParameters() - Method in class ai.djl.fasttext.FtAbstractBlock

getEpoch() - Method in class ai.djl.fasttext.FtTrainingConfig

Returns number of epochs.

getEvaluators() - Method in class ai.djl.fasttext.FtTrainingConfig

getExecutorService() - Method in class ai.djl.fasttext.FtTrainingConfig

getGroupId() - Method in class ai.djl.fasttext.zoo.FtModelZoo

getInitializers() - Method in class ai.djl.fasttext.FtTrainingConfig

getLabelPrefix() - Method in class ai.djl.fasttext.FtTrainingConfig

Return label prefix.

getLabelPrefix() - Method in class ai.djl.fasttext.zoo.nlp.textclassification.FtTextClassification

Returns the fasttext label prefix.

getLearningRate() - Method in class ai.djl.fasttext.FtTrainingConfig

Returns learning rate.

getLearningRateUpdateRate() - Method in class ai.djl.fasttext.FtTrainingConfig

Returns the rate of updates for the learning rate.

getLoss() - Method in class ai.djl.fasttext.FtTrainingConfig

Returns the loss function.

getLossFunction() - Method in class ai.djl.fasttext.FtTrainingConfig

getMaxCharLength() - Method in class ai.djl.fasttext.FtTrainingConfig

Return maximum length of char ngram.

getMaxNgramLength() - Method in class ai.djl.fasttext.FtTrainingConfig

Returns maximum length of word ngram.

getMinCharLength() - Method in class ai.djl.fasttext.FtTrainingConfig

Return minimum length of char ngram.

getMinLabelCount() - Method in class ai.djl.fasttext.FtTrainingConfig

Returns minimal number of label occurrences.

getMinWordCount() - Method in class ai.djl.fasttext.FtTrainingConfig

Return minimal number of word occurrences.

getModelFile() - Method in class ai.djl.fasttext.FtAbstractBlock

Returns the fasttext model file for the block.

getModelName() - Method in class ai.djl.fasttext.FtTrainingConfig

Return the name of the model.

getModelPath() - Method in class ai.djl.fasttext.FtModel

getModelType() - Method in class ai.djl.fasttext.jni.FtWrapper

getModelZoo() - Method in class ai.djl.fasttext.zoo.FtZooProvider

getName() - Method in class ai.djl.fasttext.FtModel

getNDManager() - Method in class ai.djl.fasttext.FtModel

getNumNegativesSampled() - Method in class ai.djl.fasttext.FtTrainingConfig

Returns number of negatives sampled.

getOptimizer() - Method in class ai.djl.fasttext.FtTrainingConfig

getOutputDir() - Method in class ai.djl.fasttext.FtTrainingConfig

Returns the output directory.

getProperties() - Method in class ai.djl.fasttext.FtModel

getProperty(String) - Method in class ai.djl.fasttext.FtModel

getSamplingThreshold() - Method in class ai.djl.fasttext.FtTrainingConfig

Returns sampling threshold.

getSupportedEngines() - Method in class ai.djl.fasttext.zoo.FtModelZoo

getThreads() - Method in class ai.djl.fasttext.FtTrainingConfig

Returns number training threads.

getTrainingListeners() - Method in class ai.djl.fasttext.FtTrainingConfig

getTrainingMode() - Method in class ai.djl.fasttext.FtTrainingConfig

Returns the @{link FtTrainingMode}.

getTrainingResult() - Method in class ai.djl.fasttext.zoo.nlp.textclassification.FtTextClassification

Returns the results of training, or null if not trained.

getWordVecSize() - Method in class ai.djl.fasttext.FtTrainingConfig

Returns size of word vectors.

getWordVector(String) - Method in class ai.djl.fasttext.jni.FtWrapper

GROUP_ID - Static variable in class ai.djl.fasttext.zoo.FtModelZoo

H

HS - Enum constant in enum class ai.djl.fasttext.FtTrainingConfig.FtLoss

L

LibUtils - Class in ai.djl.fasttext.jni

Utilities for finding the SentencePiece binary on the System.

load(InputStream, Map<String, ?>) - Method in class ai.djl.fasttext.FtModel

load(Path, String, Map<String, ?>) - Method in class ai.djl.fasttext.FtModel

loadLibrary() - Static method in class ai.djl.fasttext.jni.LibUtils

loadModel(Criteria<I, O>) - Method in class ai.djl.fasttext.zoo.nlp.textclassification.TextClassificationModelLoader

loadModel(String) - Method in class ai.djl.fasttext.jni.FtWrapper

M

modelFile - Variable in class ai.djl.fasttext.FtAbstractBlock

N

newInstance() - Static method in class ai.djl.fasttext.jni.FtWrapper

newPredictor(Translator<I, O>, Device) - Method in class ai.djl.fasttext.FtModel

newTrainer(TrainingConfig) - Method in class ai.djl.fasttext.FtModel

NS - Enum constant in enum class ai.djl.fasttext.FtTrainingConfig.FtLoss

O

optBucket(int) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional number of buckets.

optContextWindow(int) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional size of the context window.

optEpoch(int) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional number of epochs.

optLabelPrefix(String) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional label prefix.

optLearningRate(float) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional learning rate.

optLearningRateUpdateRate(int) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional rate of updates for the learning rate.

optLoss(FtTrainingConfig.FtLoss) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional loss function.

optMaxCharLength(int) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional maximum length of char ngram.

optMaxNGramLength(int) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional maximum length of word ngram.

optMinCharLength(int) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional minimum length of char ngram.

optMinLabelCount(int) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional minimal number of label occurrences.

optMinWordCount(int) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional minimal number of word occurrences.

optNumNegativesSampled(int) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional number of negatives sampled.

optSamplingThreshold(float) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional sampling threshold.

optThreads(int) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional number training threads.

optTrainingMode(FtTrainingMode) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional FtTrainingMode.

optWordVecSize(int) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the optional size of word vectors.

OVA - Enum constant in enum class ai.djl.fasttext.FtTrainingConfig.FtLoss

P

predictProba(String, int, String) - Method in class ai.djl.fasttext.jni.FtWrapper

preprocessWordToEmbed(String) - Method in class ai.djl.fasttext.zoo.nlp.word_embedding.FtWord2VecWordEmbedding

R

runCmd(String[]) - Method in class ai.djl.fasttext.jni.FtWrapper

S

save(Path, String) - Method in class ai.djl.fasttext.FtModel

setBlock(Block) - Method in class ai.djl.fasttext.FtModel

setDataType(DataType) - Method in class ai.djl.fasttext.FtModel

setModelName(String) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the name of the model.

setOutputDir(Path) - Method in class ai.djl.fasttext.FtTrainingConfig.Builder

Sets the output directory.

setProperty(String, String) - Method in class ai.djl.fasttext.FtModel

SKIPGRAM - Enum constant in enum class ai.djl.fasttext.FtTrainingMode

SOFTMAX - Enum constant in enum class ai.djl.fasttext.FtTrainingConfig.FtLoss

SUPERVISED - Enum constant in enum class ai.djl.fasttext.FtTrainingMode

T

textClassification(FtTrainingConfig, RawDataset<Path>) - Static method in class ai.djl.fasttext.TrainFastText

Trains a fastText Application.NLP.TEXT_CLASSIFICATION model.

TextClassificationModelLoader - Class in ai.djl.fasttext.zoo.nlp.textclassification

Model loader for fastText cooking stackexchange models.

TextClassificationModelLoader(Repository) - Constructor for class ai.djl.fasttext.zoo.nlp.textclassification.TextClassificationModelLoader

Creates the Model loader from the given repository.

toCommand(String) - Method in class ai.djl.fasttext.FtTrainingConfig

Returns the fastText command in an array.

toString() - Method in class ai.djl.fasttext.FtModel

toWordEmbedding() - Method in class ai.djl.fasttext.zoo.nlp.textclassification.FtTextClassification

Converts the block into the equivalent FtWordEmbeddingBlock.

TrainFastText - Class in ai.djl.fasttext

A utility to aggregate options for training with fasttext.

U

unembedWord(NDArray) - Method in class ai.djl.fasttext.zoo.nlp.word_embedding.FtWord2VecWordEmbedding

unloadModel() - Method in class ai.djl.fasttext.jni.FtWrapper

V

valueOf(String) - Static method in enum class ai.djl.fasttext.FtTrainingConfig.FtLoss

Returns the enum constant of this class with the specified name.

valueOf(String) - Static method in enum class ai.djl.fasttext.FtTrainingMode

Returns the enum constant of this class with the specified name.

values() - Static method in enum class ai.djl.fasttext.FtTrainingConfig.FtLoss

Returns an array containing the constants of this enum class, in the order they are declared.

values() - Static method in enum class ai.djl.fasttext.FtTrainingMode

Returns an array containing the constants of this enum class, in the order they are declared.

vocabularyContains(String) - Method in class ai.djl.fasttext.zoo.nlp.word_embedding.FtWord2VecWordEmbedding

A B C D E F G H L M N O P R S T U V
All Classes and Interfaces|All Packages|Constant Field Values