Uses of Class
org.tribuo.interop.tensorflow.TensorMap
Package
Description
Provides an interface to TensorFlow, allowing the training of non-sequential models using any supported
Tribuo output type.
Provides an interface for working with TensorFlow sequence models, using Tribuo's
SequenceModel
abstraction.-
Uses of TensorMap in org.tribuo.interop.tensorflow
Modifier and TypeMethodDescriptionDenseFeatureConverter.convert
(List<? extends Example<?>> examples, ImmutableFeatureMap featureIDMap) DenseFeatureConverter.convert
(Example<?> example, ImmutableFeatureMap featureIDMap) FeatureConverter.convert
(List<? extends Example<?>> example, ImmutableFeatureMap featureIDMap) FeatureConverter.convert
(Example<?> example, ImmutableFeatureMap featureIDMap) ImageConverter.convert
(List<? extends Example<?>> examples, ImmutableFeatureMap featureIDMap) Transform implicitly pads unseen values with zero.ImageConverter.convert
(Example<?> example, ImmutableFeatureMap featureIDMap) Transform implicitly pads unseen values with zero.protected TensorMap
TensorFlowFrozenExternalModel.convertFeatures
(SparseVector input) protected TensorMap
TensorFlowSavedModelExternalModel.convertFeatures
(SparseVector input) protected TensorMap
TensorFlowFrozenExternalModel.convertFeaturesList
(List<SparseVector> input) protected TensorMap
TensorFlowSavedModelExternalModel.convertFeaturesList
(List<SparseVector> input) protected TensorMap
TensorFlowSavedModelExternalModel.externalPrediction
(TensorMap input) Runs the session to make a prediction.Modifier and TypeMethodDescriptionprotected List<Prediction<T>>
TensorFlowSavedModelExternalModel.convertOutput
(TensorMap output, int[] numValidFeatures, List<Example<T>> examples) Converts a tensor into a prediction.protected Prediction<T>
TensorFlowSavedModelExternalModel.convertOutput
(TensorMap output, int numValidFeatures, Example<T> example) Converts a tensor into a prediction.protected org.tensorflow.Tensor
TensorFlowFrozenExternalModel.externalPrediction
(TensorMap input) Runs the session to make a prediction.protected TensorMap
TensorFlowSavedModelExternalModel.externalPrediction
(TensorMap input) Runs the session to make a prediction. -
Uses of TensorMap in org.tribuo.interop.tensorflow.sequence
Modifier and TypeMethodDescriptionSequenceFeatureConverter.encode
(List<? extends SequenceExample<?>> batch, ImmutableFeatureMap featureMap) Encodes a batch of examples as a feed dict.SequenceFeatureConverter.encode
(SequenceExample<?> example, ImmutableFeatureMap featureMap) Encodes an example as a feed dict.SequenceOutputConverter.encode
(List<SequenceExample<T>> batch, ImmutableOutputInfo<T> labelMap) Encodes a batch of labels as a feed dict.SequenceOutputConverter.encode
(SequenceExample<T> example, ImmutableOutputInfo<T> labelMap) Encodes an example's label as a feed dict.protected TensorMap
TensorFlowSequenceTrainer.getHyperparameterFeed()
Build any necessary non-data parameter tensors.