com.spotify.zoltar.tf

Interface TensorFlowPredictFn<InputT,VectorT,ValueT>

Type Parameters:

InputT - type of the raw input to the feature extraction.

VectorT - type of the feature extraction output.

ValueT - type of the prediction result.

All Superinterfaces:

PredictFns.AsyncPredictFn<TensorFlowModel,InputT,VectorT,ValueT>

Functional Interface:

This is a functional interface and can therefore be used as the assignment target for a lambda expression or method reference.

@FunctionalInterface
public interface TensorFlowPredictFn<InputT,VectorT,ValueT>
extends PredictFns.AsyncPredictFn<TensorFlowModel,InputT,VectorT,ValueT>

TensorFlow flavor of PredictFns.AsyncPredictFn using TensorFlowModel.

Method Summary

Modifier and Type	Method and Description
static <InputT,ValueT> TensorFlowPredictFn<InputT,org.tensorflow.example.Example,ValueT>	example(Function<Map<String,JTensor>,ValueT> outTensorExtractor, String... fetchOps) TensorFlow Example prediction function.
static <InputT,ValueT> TensorFlowPredictFn<InputT,List<org.tensorflow.example.Example>,ValueT>	exampleBatch(Function<Map<String,JTensor>,ValueT> outTensorExtractor, String... fetchOps) TensorFlow Example prediction function.

Methods inherited from interface com.spotify.zoltar.PredictFns.AsyncPredictFn

apply, lift, with

Method Detail

example

static <InputT,ValueT> TensorFlowPredictFn<InputT,org.tensorflow.example.Example,ValueT> example(Function<Map<String,JTensor>,ValueT> outTensorExtractor,
                                                                                                 String... fetchOps)

TensorFlow Example prediction function.

Parameters:

outTensorExtractor - Function to extract the output value from JTensor's

fetchOps - operations to fetch.

exampleBatch

static <InputT,ValueT> TensorFlowPredictFn<InputT,List<org.tensorflow.example.Example>,ValueT> exampleBatch(Function<Map<String,JTensor>,ValueT> outTensorExtractor,
                                                                                                            String... fetchOps)

TensorFlow Example prediction function.

Parameters:

outTensorExtractor - Function to extract the output value from JTensor's

fetchOps - operations to fetch.