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

Class PytorchQAComputeJob<I extends org.gridgain.ml.model.MlSimpleJobParameters,O>

java.lang.Object

org.gridgain.ml.compute.MlComputeJob<I,O>

org.apache.ignite.example.ml.pytorchtranslator.PytorchQAComputeJob<I,O>

All Implemented Interfaces:

org.apache.ignite.compute.ComputeJob<I,O>

public class PytorchQAComputeJob<I extends org.gridgain.ml.model.MlSimpleJobParameters,O>
extends org.gridgain.ml.compute.MlComputeJob<I,O>

Compute job for PyTorch QA ML predictions.

Constructor Summary

Constructors

Constructor	Description
PytorchQAComputeJob()

Method Summary

Modifier and Type	Method	Description
org.apache.ignite.marshalling.Marshaller<I,byte[]>	inputMarshaller()
CompletableFuture<O>	predictAsync(org.apache.ignite.compute.JobExecutionContext context, I arg)
org.apache.ignite.marshalling.Marshaller<O,byte[]>	resultMarshaller()

Methods inherited from class org.gridgain.ml.compute.MlComputeJob

executeAsync, updateModelPathFromDeploymentUnits

Methods inherited from class java.lang.Object

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

Constructor Details

PytorchQAComputeJob

public PytorchQAComputeJob()

Method Details

predictAsync

public CompletableFuture<O> predictAsync(org.apache.ignite.compute.JobExecutionContext context,
 I arg)

Specified by:

predictAsync in class org.gridgain.ml.compute.MlComputeJob<I extends org.gridgain.ml.model.MlSimpleJobParameters,O>

inputMarshaller

public org.apache.ignite.marshalling.Marshaller<I,byte[]> inputMarshaller()

Specified by:

inputMarshaller in interface org.apache.ignite.compute.ComputeJob<I extends org.gridgain.ml.model.MlSimpleJobParameters,O>

Overrides:

inputMarshaller in class org.gridgain.ml.compute.MlComputeJob<I extends org.gridgain.ml.model.MlSimpleJobParameters,O>

resultMarshaller

public org.apache.ignite.marshalling.Marshaller<O,byte[]> resultMarshaller()