Package org.apache.beam.sdk.ml.inference.remote

Class RemoteInference.Invoke<InputT extends BaseInput,OutputT extends BaseResponse>

java.lang.Object

org.apache.beam.sdk.transforms.PTransform<org.apache.beam.sdk.values.PCollection<InputT>,org.apache.beam.sdk.values.PCollection<java.lang.Iterable<PredictionResult<InputT,OutputT>>>>

org.apache.beam.sdk.ml.inference.remote.RemoteInference.Invoke<InputT,OutputT>

All Implemented Interfaces:

java.io.Serializable, org.apache.beam.sdk.transforms.display.HasDisplayData

Enclosing class:

RemoteInference

public abstract static class RemoteInference.Invoke<InputT extends BaseInput,OutputT extends BaseResponse>
extends org.apache.beam.sdk.transforms.PTransform<org.apache.beam.sdk.values.PCollection<InputT>,org.apache.beam.sdk.values.PCollection<java.lang.Iterable<PredictionResult<InputT,OutputT>>>>

See Also:

Serialized Form

Field Summary

Fields inherited from class org.apache.beam.sdk.transforms.PTransform

annotations, displayData, name, resourceHints

Constructor Summary

Constructors

Constructor	Description
Invoke()

Method Summary

Modifier and Type	Method	Description
org.apache.beam.sdk.values.PCollection<java.lang.Iterable<PredictionResult<InputT,OutputT>>>	expand(org.apache.beam.sdk.values.PCollection<InputT> input)
RemoteInference.Invoke<InputT,OutputT>	handler(java.lang.Class<? extends BaseModelHandler<?,InputT,OutputT>> modelHandler)	Model handler class for inference.
RemoteInference.Invoke<InputT,OutputT>	withParameters(BaseModelParameters modelParameters)	Configures the parameters for model initialization.

Methods inherited from class org.apache.beam.sdk.transforms.PTransform

addAnnotation, compose, compose, getAdditionalInputs, getAnnotations, getDefaultOutputCoder, getDefaultOutputCoder, getDefaultOutputCoder, getKindString, getName, getResourceHints, populateDisplayData, setDisplayData, setResourceHints, toString, validate, validate

Methods inherited from class java.lang.Object

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

Constructor Detail

Invoke

public Invoke()

Method Detail

handler

public RemoteInference.Invoke<InputT,OutputT> handler(java.lang.Class<? extends BaseModelHandler<?,InputT,OutputT>> modelHandler)

Model handler class for inference.

withParameters

public RemoteInference.Invoke<InputT,OutputT> withParameters(BaseModelParameters modelParameters)

Configures the parameters for model initialization.

expand

public org.apache.beam.sdk.values.PCollection<java.lang.Iterable<PredictionResult<InputT,OutputT>>> expand(org.apache.beam.sdk.values.PCollection<InputT> input)

Specified by:

expand in class org.apache.beam.sdk.transforms.PTransform<org.apache.beam.sdk.values.PCollection<InputT extends BaseInput>,org.apache.beam.sdk.values.PCollection<java.lang.Iterable<PredictionResult<InputT extends BaseInput,OutputT extends BaseResponse>>>>