Package ai.djl.pytorch.zoo.nlp.qa

Class PtBertQATranslator

java.lang.Object
ai.djl.modality.nlp.translator.QATranslator
ai.djl.pytorch.zoo.nlp.qa.PtBertQATranslator
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 PtBertQATranslator extends ai.djl.modality.nlp.translator.QATranslator
The Translator for PyTorch Question Answering model.

  • Nested Class Summary

    Nested Classes
    Modifier and Type
    Class
    Description
    static class 
    PtBertQATranslator.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 PtBertQATranslator.Builder
    builder()
    Creates a builder to build a PtBertQATranslator.
    static PtBertQATranslator.Builder
    builder(Map<String,?> arguments)
    Creates a builder to build a PtSSDTranslatorBuilder with specified arguments.
    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 ai.djl.modality.nlp.translator.QATranslator

    getBatchifier

    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

    • 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 PtBertQATranslator.Builder builder()
      Creates a builder to build a PtBertQATranslator.
      Returns:
      a new builder

    • builder

      public static PtBertQATranslator.Builder builder(Map<String,?> arguments)
      Creates a builder to build a PtSSDTranslatorBuilder with specified arguments.
      Parameters:
      arguments - arguments to specify builder options
      Returns:
      a new builder