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

Class MxBertQATranslator

java.lang.Object
ai.djl.modality.nlp.translator.QATranslator
ai.djl.mxnet.zoo.nlp.qa.MxBertQATranslator
All Implemented Interfaces:
ai.djl.translate.PostProcessor<String>, ai.djl.translate.PreProcessor<ai.djl.modality.nlp.qa.QAInput>, ai.djl.translate.Translator<ai.djl.modality.nlp.qa.QAInput,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

    Modifier and Type
    Method
    Description
    static MxBertQATranslator.Builder
    builder()
    Creates a builder to build a MxBertQATranslator.
    static MxBertQATranslator.Builder
    builder(Map<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)
    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

    Methods inherited from interface ai.djl.translate.Translator

    batchProcessInput, batchProcessOutput, getExpansions

  • Method Details

    • prepare

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

    • getBatchifier

      public ai.djl.translate.Batchifier getBatchifier()
      Specified by:
      getBatchifier in interface ai.djl.translate.Translator<ai.djl.modality.nlp.qa.QAInput,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 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(Map<String,?> arguments)
      Creates a builder to build a MxBertQATranslator.
      Parameters:
      arguments - the models' arguments
      Returns:
      a new builder