Package ai.djl.huggingface.translator

Class FillMaskTranslator

java.lang.Object
ai.djl.huggingface.translator.FillMaskTranslator
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 FillMaskTranslator extends Object implements ai.djl.translate.Translator<String,ai.djl.modality.Classifications>
The translator for Huggingface fill mask model.

  • Method Details

    • getBatchifier

      public ai.djl.translate.Batchifier getBatchifier()
      Specified by:
      getBatchifier in interface ai.djl.translate.Translator<String,ai.djl.modality.Classifications>

    • processInput

      public ai.djl.ndarray.NDList processInput(ai.djl.translate.TranslatorContext ctx, String input) throws ai.djl.translate.TranslateException
      Specified by:
      processInput in interface ai.djl.translate.PreProcessor<String>
      Throws:
      ai.djl.translate.TranslateException

    • batchProcessInput

      public ai.djl.ndarray.NDList batchProcessInput(ai.djl.translate.TranslatorContext ctx, List<String> inputs) throws ai.djl.translate.TranslateException
      Specified by:
      batchProcessInput in interface ai.djl.translate.Translator<String,ai.djl.modality.Classifications>
      Throws:
      ai.djl.translate.TranslateException

    • 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>

    • batchProcessOutput

      public List<ai.djl.modality.Classifications> batchProcessOutput(ai.djl.translate.TranslatorContext ctx, ai.djl.ndarray.NDList list)
      Specified by:
      batchProcessOutput in interface ai.djl.translate.Translator<String,ai.djl.modality.Classifications>

    • builder

      public static FillMaskTranslator.Builder builder(HuggingFaceTokenizer tokenizer)
      Creates a builder to build a FillMaskTranslator.
      Parameters:
      tokenizer - the tokenizer
      Returns:
      a new builder

    • builder

      public static FillMaskTranslator.Builder builder(HuggingFaceTokenizer tokenizer, Map<String,?> arguments)
      Creates a builder to build a FillMaskTranslator.
      Parameters:
      tokenizer - the tokenizer
      arguments - the models' arguments
      Returns:
      a new builder