Package ai.djl.huggingface.translator

Class QuestionAnsweringBatchTranslator

  • All Implemented Interfaces:
    ai.djl.translate.NoBatchifyTranslator<ai.djl.modality.nlp.qa.QAInput[],​java.lang.String[]>, 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 QuestionAnsweringBatchTranslator
extends java.lang.Object
implements ai.djl.translate.NoBatchifyTranslator<ai.djl.modality.nlp.qa.QAInput[],​java.lang.String[]>
    The translator for Huggingface question answering model.

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      ai.djl.ndarray.NDList processInput​(ai.djl.translate.TranslatorContext ctx, ai.djl.modality.nlp.qa.QAInput[] inputs)
      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

      • Methods inherited from interface ai.djl.translate.NoBatchifyTranslator

        getBatchifier

      • Methods inherited from interface ai.djl.translate.Translator

        getExpansions, prepare, toBatchTranslator, toBatchTranslator

    • Method Detail

      • processInput

        public ai.djl.ndarray.NDList processInput​(ai.djl.translate.TranslatorContext ctx,
                                          ai.djl.modality.nlp.qa.QAInput[] inputs)
        Specified by:
        processInput in interface ai.djl.translate.PreProcessor<ai.djl.modality.nlp.qa.QAInput[]>

      • processOutput

        public java.lang.String[] processOutput​(ai.djl.translate.TranslatorContext ctx,
                                        ai.djl.ndarray.NDList list)
        Specified by:
        processOutput in interface ai.djl.translate.PostProcessor<java.lang.String[]>