Package ai.djl.huggingface.translator

Class TokenClassificationTranslator

java.lang.Object

ai.djl.huggingface.translator.TokenClassificationTranslator

All Implemented Interfaces:

ai.djl.translate.PostProcessor<ai.djl.modality.nlp.translator.NamedEntity[]>, ai.djl.translate.PreProcessor<java.lang.String>, ai.djl.translate.Translator<java.lang.String,ai.djl.modality.nlp.translator.NamedEntity[]>

public class TokenClassificationTranslator
extends java.lang.Object
implements ai.djl.translate.Translator<java.lang.String,ai.djl.modality.nlp.translator.NamedEntity[]>

The translator for Huggingface token classification model.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	TokenClassificationTranslator.Builder	The builder for token classification translator.

Method Summary

Modifier and Type	Method	Description
static TokenClassificationTranslator.Builder	builder(HuggingFaceTokenizer tokenizer)	Creates a builder to build a TokenClassificationTranslator.
static TokenClassificationTranslator.Builder	builder(HuggingFaceTokenizer tokenizer, java.util.Map<java.lang.String,?> arguments)	Creates a builder to build a TokenClassificationTranslator.
ai.djl.translate.Batchifier	getBatchifier()
void	prepare(ai.djl.translate.TranslatorContext ctx)
ai.djl.ndarray.NDList	processInput(ai.djl.translate.TranslatorContext ctx, java.lang.String input)
ai.djl.modality.nlp.translator.NamedEntity[]	processOutput(ai.djl.translate.TranslatorContext ctx, ai.djl.ndarray.NDList list)
TokenClassificationBatchTranslator	toBatchTranslator(ai.djl.translate.Batchifier batchifier)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface ai.djl.translate.Translator

getExpansions, toBatchTranslator

Method Detail

getBatchifier

public ai.djl.translate.Batchifier getBatchifier()

Specified by:

getBatchifier in interface ai.djl.translate.Translator<java.lang.String,ai.djl.modality.nlp.translator.NamedEntity[]>

prepare

public void prepare(ai.djl.translate.TranslatorContext ctx)
             throws java.io.IOException

Specified by:

prepare in interface ai.djl.translate.Translator<java.lang.String,ai.djl.modality.nlp.translator.NamedEntity[]>

Throws:

java.io.IOException

processInput

public ai.djl.ndarray.NDList processInput(ai.djl.translate.TranslatorContext ctx,
                                          java.lang.String input)

Specified by:

processInput in interface ai.djl.translate.PreProcessor<java.lang.String>

processOutput

public ai.djl.modality.nlp.translator.NamedEntity[] processOutput(ai.djl.translate.TranslatorContext ctx,
                                                                  ai.djl.ndarray.NDList list)

Specified by:

processOutput in interface ai.djl.translate.PostProcessor<ai.djl.modality.nlp.translator.NamedEntity[]>

toBatchTranslator

public TokenClassificationBatchTranslator toBatchTranslator(ai.djl.translate.Batchifier batchifier)

Specified by:

toBatchTranslator in interface ai.djl.translate.Translator<java.lang.String,ai.djl.modality.nlp.translator.NamedEntity[]>

builder

public static TokenClassificationTranslator.Builder builder(HuggingFaceTokenizer tokenizer)

Creates a builder to build a TokenClassificationTranslator.

Parameters:

tokenizer - the tokenizer

Returns:

a new builder

builder

public static TokenClassificationTranslator.Builder builder(HuggingFaceTokenizer tokenizer,
                                                            java.util.Map<java.lang.String,?> arguments)

Creates a builder to build a TokenClassificationTranslator.

Parameters:

tokenizer - the tokenizer

arguments - the models' arguments

Returns:

a new builder