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

Class PtDistilBertTranslator

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

  • Constructor Summary

    Constructors
    Constructor
    Description
    PtDistilBertTranslator()
     

  • Method Summary

    Modifier and Type
    Method
    Description
    void
    prepare(ai.djl.translate.TranslatorContext ctx)
    ai.djl.ndarray.NDList
    processInput(ai.djl.translate.TranslatorContext ctx, 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

    batchProcessInput, batchProcessOutput, getBatchifier, getExpansions

  • Constructor Details

    • PtDistilBertTranslator

      public PtDistilBertTranslator()

  • Method Details

    • prepare

      public void prepare(ai.djl.translate.TranslatorContext ctx) throws IOException
      Specified by:
      prepare in interface ai.djl.translate.Translator<String,ai.djl.modality.Classifications>
      Throws:
      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, String input)
      Specified by:
      processInput in interface ai.djl.translate.PreProcessor<String>