Class TokenClassificationTranslator.Builder

  • Enclosing class:
    TokenClassificationTranslator

    public static final class TokenClassificationTranslator.Builder
    extends java.lang.Object
    The builder for token classification translator.
    • Method Detail

      • 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​(java.util.Map<java.lang.String,​?> arguments)
        Configures the builder with the model arguments.
        Parameters:
        arguments - the model arguments
      • build

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