Package ai.djl.pytorch.zoo.nlp.sentimentanalysis

Class PtDistilBertTranslator

  • All Implemented Interfaces:
    ai.djl.translate.PostProcessor<ai.djl.modality.Classifications>, ai.djl.translate.PreProcessor<java.lang.String>, ai.djl.translate.Translator<java.lang.String,​ai.djl.modality.Classifications>
    public class PtDistilBertTranslator
extends java.lang.Object
implements ai.djl.translate.Translator<java.lang.String,​ai.djl.modality.Classifications>
    The Translator for PyTorch Sentiment Analysis model.

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      void prepare​(ai.djl.translate.TranslatorContext ctx)
      ai.djl.ndarray.NDList processInput​(ai.djl.translate.TranslatorContext ctx, java.lang.String input)
      ai.djl.modality.Classifications 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

        getBatchifier, getExpansions, toBatchTranslator, toBatchTranslator

    • Constructor Detail

      • PtDistilBertTranslator

        public PtDistilBertTranslator()

    • Method Detail

      • prepare

        public void prepare​(ai.djl.translate.TranslatorContext ctx)
             throws java.io.IOException
        Specified by:
        prepare in interface ai.djl.translate.Translator<java.lang.String,​ai.djl.modality.Classifications>
        Throws:
        java.io.IOException

      • processOutput

        public ai.djl.modality.Classifications processOutput​(ai.djl.translate.TranslatorContext ctx,
                                                     ai.djl.ndarray.NDList list)
        Specified by:
        processOutput in interface ai.djl.translate.PostProcessor<ai.djl.modality.Classifications>

      • processInput

        public ai.djl.ndarray.NDList processInput​(ai.djl.translate.TranslatorContext ctx,
                                          java.lang.String input)
        Specified by:
        processInput in interface ai.djl.translate.PreProcessor<java.lang.String>