Package org.apache.ignite.example.ml.pytorchtranslator

Class PytorchQATranslator

java.lang.Object

org.apache.ignite.example.ml.pytorchtranslator.PytorchQATranslator

All Implemented Interfaces:

ai.djl.translate.PostProcessor<String>, ai.djl.translate.PreProcessor<PytorchQAInput>, ai.djl.translate.Translator<PytorchQAInput,String>

public class PytorchQATranslator
extends Object
implements ai.djl.translate.Translator<PytorchQAInput,String>

Translator for PyTorch-based Question Answering models.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final class	PytorchQATranslator.Builder

Method Summary

Modifier and Type	Method	Description
ai.djl.ndarray.NDList	batchProcessInput(ai.djl.translate.TranslatorContext ctx, List<PytorchQAInput> inputs)
List<String>	batchProcessOutput(ai.djl.translate.TranslatorContext ctx, ai.djl.ndarray.NDList list)
static PytorchQATranslator.Builder	builder(ai.djl.huggingface.tokenizers.HuggingFaceTokenizer tokenizer)
static PytorchQATranslator.Builder	builder(ai.djl.huggingface.tokenizers.HuggingFaceTokenizer tokenizer, Map<String,?> arguments)
ai.djl.translate.Batchifier	getBatchifier()
ai.djl.ndarray.NDList	processInput(ai.djl.translate.TranslatorContext ctx, PytorchQAInput input)
String	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.Translator

getExpansions, prepare

Method Details

getBatchifier

public ai.djl.translate.Batchifier getBatchifier()

Specified by:

getBatchifier in interface ai.djl.translate.Translator<PytorchQAInput,String>

processInput

public ai.djl.ndarray.NDList processInput(ai.djl.translate.TranslatorContext ctx,
 PytorchQAInput input)

Specified by:

processInput in interface ai.djl.translate.PreProcessor<PytorchQAInput>

batchProcessInput

public ai.djl.ndarray.NDList batchProcessInput(ai.djl.translate.TranslatorContext ctx,
 List<PytorchQAInput> inputs)

Specified by:

batchProcessInput in interface ai.djl.translate.Translator<PytorchQAInput,String>

processOutput

public String processOutput(ai.djl.translate.TranslatorContext ctx,
 ai.djl.ndarray.NDList list)

Specified by:

processOutput in interface ai.djl.translate.PostProcessor<String>

batchProcessOutput

public List<String> batchProcessOutput(ai.djl.translate.TranslatorContext ctx,
 ai.djl.ndarray.NDList list)

Specified by:

batchProcessOutput in interface ai.djl.translate.Translator<PytorchQAInput,String>

builder

public static PytorchQATranslator.Builder builder(ai.djl.huggingface.tokenizers.HuggingFaceTokenizer tokenizer)

builder

public static PytorchQATranslator.Builder builder(ai.djl.huggingface.tokenizers.HuggingFaceTokenizer tokenizer,
 Map<String,?> arguments)