Class TextClassificationTranslator.Builder

  • Enclosing class:
    TextClassificationTranslator

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

      • optBatchifier

        public TextClassificationTranslator.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 TextClassificationTranslator build()
                                           throws java.io.IOException
        Builds the translator.
        Returns:
        the new translator
        Throws:
        java.io.IOException - if I/O error occurs