Package org.apache.ignite.example.ml.tensorflowtranslator

Class SentenceEncoderTranslator

java.lang.Object
org.apache.ignite.example.ml.tensorflowtranslator.SentenceEncoderTranslator
All Implemented Interfaces:
ai.djl.translate.NoBatchifyTranslator<String[],float[][]>, ai.djl.translate.PostProcessor<float[][]>, ai.djl.translate.PreProcessor<String[]>, ai.djl.translate.Translator<String[],float[][]>
public class SentenceEncoderTranslator extends Object implements ai.djl.translate.NoBatchifyTranslator<String[],float[][]>

  • Constructor Summary

    Constructors
    Constructor
    Description
    SentenceEncoderTranslator()
     

  • Method Summary

    Modifier and Type
    Method
    Description
    ai.djl.ndarray.NDList
    processInput(ai.djl.translate.TranslatorContext ctx, String[] inputs)
     
    float[][]
    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

    batchProcessInput, batchProcessOutput, getExpansions, prepare

  • Constructor Details

    • SentenceEncoderTranslator

      public SentenceEncoderTranslator()

  • Method Details

    • processInput

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

    • processOutput

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