Package ai.djl.huggingface.translator
Class TextClassificationTranslator.Builder
- java.lang.Object
-
- ai.djl.huggingface.translator.TextClassificationTranslator.Builder
-
- Enclosing class:
- TextClassificationTranslator
public static final class TextClassificationTranslator.Builder extends java.lang.ObjectThe builder for token classification translator.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description TextClassificationTranslatorbuild()Builds the translator.voidconfigure(java.util.Map<java.lang.String,?> arguments)Configures the builder with the model arguments.TextClassificationTranslator.BuilderoptBatchifier(ai.djl.translate.Batchifier batchifier)Sets theBatchifierfor theTranslator.
-
-
-
Method Detail
-
optBatchifier
public TextClassificationTranslator.Builder optBatchifier(ai.djl.translate.Batchifier batchifier)
Sets theBatchifierfor theTranslator.- Parameters:
batchifier- true to include token types- Returns:
- this builder
-
configure
public void configure(java.util.Map<java.lang.String,?> arguments)
Configures the builder with the model arguments.- Parameters:
arguments- the model arguments
-
build
public TextClassificationTranslator build() throws java.io.IOException
Builds the translator.- Returns:
- the new translator
- Throws:
java.io.IOException- if I/O error occurs
-
-