Package org.apache.camel.component.djl.model.nlp

Class AbstractNlpZooPredictor<T>

java.lang.Object

org.apache.camel.component.djl.model.AbstractPredictor

org.apache.camel.component.djl.model.nlp.AbstractNlpZooPredictor<T>

Direct Known Subclasses:

ZooFillMaskPredictor, ZooMachineTranslationPredictor, ZooMultipleChoicePredictor, ZooSentimentAnalysisPredictor, ZooTextClassificationPredictor, ZooTextEmbeddingPredictor, ZooTextGenerationPredictor, ZooTokenClassificationPredictor, ZooWordEmbeddingPredictor

public abstract class AbstractNlpZooPredictor<T>
extends AbstractPredictor

Field Summary

Fields

Modifier and Type	Field	Description
protected ai.djl.repository.zoo.ZooModel<String,T>	model

Constructor Summary

Constructors

Constructor	Description
AbstractNlpZooPredictor(DJLEndpoint endpoint)

Method Summary

Modifier and Type	Method	Description
protected T	predict(org.apache.camel.Exchange exchange, String input)
void	process(org.apache.camel.Exchange exchange)

Methods inherited from class org.apache.camel.component.djl.model.AbstractPredictor

getEndpoint

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

model

protected ai.djl.repository.zoo.ZooModel<String,T> model

Constructor Details

AbstractNlpZooPredictor

public AbstractNlpZooPredictor(DJLEndpoint endpoint)

Method Details

process

public void process(org.apache.camel.Exchange exchange)

Specified by:

process in class AbstractPredictor

predict

protected T predict(org.apache.camel.Exchange exchange,
 String input)