Package org.apache.ignite.example.ml.custom

Class CustomTranslatorFactory

java.lang.Object

org.apache.ignite.example.ml.custom.CustomTranslatorFactory

All Implemented Interfaces:

ai.djl.translate.TranslatorFactory, Serializable

public class CustomTranslatorFactory
extends Object
implements ai.djl.translate.TranslatorFactory, Serializable

Factory for creating instances of CustomTranslator.

Initializes the Hugging Face tokenizer and provides translator instances that convert CustomInput objects into CustomOutput predictions.

See Also:

• Serialized Form

Constructor Summary

Constructors

Constructor	Description
CustomTranslatorFactory()

Method Summary

Modifier and Type	Method	Description
Set<ai.djl.util.Pair<Type,Type>>	getSupportedTypes()
<I, O> ai.djl.translate.Translator<I,O>	newInstance(Class<I> input, Class<O> output, ai.djl.Model model, Map<String,?> arguments)

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.TranslatorFactory

isSupported

Constructor Details

CustomTranslatorFactory

public CustomTranslatorFactory()

Method Details

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

getSupportedTypes

public Set<ai.djl.util.Pair<Type,Type>> getSupportedTypes()

Specified by:

getSupportedTypes in interface ai.djl.translate.TranslatorFactory