Package ai.djl.huggingface.translator

Class TokenClassificationTranslator.Builder

java.lang.Object
ai.djl.huggingface.translator.TokenClassificationTranslator.Builder
Enclosing class:
TokenClassificationTranslator
public static final class TokenClassificationTranslator.Builder extends Object
The builder for token classification translator.

  • Method Details

    • optIncludeTokenTypes

      public TokenClassificationTranslator.Builder optIncludeTokenTypes(boolean includeTokenTypes)
      Sets if include token types for the Translator.
      Parameters:
      includeTokenTypes - true to include token types
      Returns:
      this builder

    • optSoftmax

      public TokenClassificationTranslator.Builder optSoftmax(boolean softmax)
      Sets if implement softmax operation for the Translator.
      Parameters:
      softmax - true to implement softmax to model output result
      Returns:
      this builder

    • optBatchifier

      public TokenClassificationTranslator.Builder optBatchifier(ai.djl.translate.Batchifier batchifier)
      Sets the Batchifier for the Translator.
      Parameters:
      batchifier - true to include token types
      Returns:
      this builder

    • configure

      public void configure(Map<String,?> arguments)
      Configures the builder with the model arguments.
      Parameters:
      arguments - the model arguments

    • build

      public TokenClassificationTranslator build() throws IOException
      Builds the translator.
      Returns:
      the new translator
      Throws:
      IOException - if I/O error occurs