Class TextClassificationTranslatorFactory

java.lang.Object
ai.djl.huggingface.translator.TextClassificationTranslatorFactory
All Implemented Interfaces:
ai.djl.translate.TranslatorFactory, Serializable

public class TextClassificationTranslatorFactory extends Object implements ai.djl.translate.TranslatorFactory, Serializable
A TranslatorFactory that creates a TextClassificationTranslator instance.
See Also:
  • Constructor Details

    • TextClassificationTranslatorFactory

      public TextClassificationTranslatorFactory()
  • Method Details

    • getSupportedTypes

      public Set<ai.djl.util.Pair<Type,Type>> getSupportedTypes()
      Specified by:
      getSupportedTypes in interface ai.djl.translate.TranslatorFactory
    • newInstance

      public <I, O> ai.djl.translate.Translator<I,O> newInstance(Class<I> input, Class<O> output, ai.djl.Model model, Map<String,?> arguments) throws ai.djl.translate.TranslateException
      Specified by:
      newInstance in interface ai.djl.translate.TranslatorFactory
      Throws:
      ai.djl.translate.TranslateException