Package ai.djl.mxnet.zoo.nlp.qa

Class MxBertQATranslator

  • All Implemented Interfaces:
    ai.djl.translate.PostProcessor<java.lang.String>, ai.djl.translate.PreProcessor<ai.djl.modality.nlp.qa.QAInput>, ai.djl.translate.Translator<ai.djl.modality.nlp.qa.QAInput,​java.lang.String>
    public class MxBertQATranslator
extends ai.djl.modality.nlp.translator.QATranslator
    The translator for MXNet BERT QA model.

    • Nested Class Summary

      Nested Classes 
      Modifier and Type Class Description
      static class  MxBertQATranslator.Builder
      The builder for Bert QA translator.

      • Nested classes/interfaces inherited from class ai.djl.modality.nlp.translator.QATranslator

        ai.djl.modality.nlp.translator.QATranslator.BaseBuilder<T extends ai.djl.modality.nlp.translator.QATranslator.BaseBuilder>

    • Field Summary

      • Fields inherited from class ai.djl.modality.nlp.translator.QATranslator

        batchifier, includeTokenTypes, locale, maxLabels, maxLength, padding, tokenizerName, toLowerCase, truncation, vocab

    • Method Summary

      All Methods Static Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      static MxBertQATranslator.Builder builder()
      Creates a builder to build a MxBertQATranslator.
      static MxBertQATranslator.Builder builder​(java.util.Map<java.lang.String,​?> arguments)
      Creates a builder to build a MxBertQATranslator.
      ai.djl.translate.Batchifier getBatchifier()
      void prepare​(ai.djl.translate.TranslatorContext ctx)
      ai.djl.ndarray.NDList processInput​(ai.djl.translate.TranslatorContext ctx, ai.djl.modality.nlp.qa.QAInput input)
      java.lang.String processOutput​(ai.djl.translate.TranslatorContext ctx, ai.djl.ndarray.NDList list)

      • Methods inherited from class java.lang.Object

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

    • Method Detail

      • prepare

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

      • getBatchifier

        public ai.djl.translate.Batchifier getBatchifier()
        Specified by:
        getBatchifier in interface ai.djl.translate.Translator<ai.djl.modality.nlp.qa.QAInput,​java.lang.String>
        Overrides:
        getBatchifier in class ai.djl.modality.nlp.translator.QATranslator

      • processInput

        public ai.djl.ndarray.NDList processInput​(ai.djl.translate.TranslatorContext ctx,
                                          ai.djl.modality.nlp.qa.QAInput input)

      • processOutput

        public java.lang.String processOutput​(ai.djl.translate.TranslatorContext ctx,
                                      ai.djl.ndarray.NDList list)

      • builder

        public static MxBertQATranslator.Builder builder()
        Creates a builder to build a MxBertQATranslator.
        Returns:
        a new builder

      • builder

        public static MxBertQATranslator.Builder builder​(java.util.Map<java.lang.String,​?> arguments)
        Creates a builder to build a MxBertQATranslator.
        Parameters:
        arguments - the models' arguments
        Returns:
        a new builder