Package | Description |
---|---|
org.tribuo.interop.tensorflow |
Provides an interface to TensorFlow, allowing the training of non-sequential models using any supported
Tribuo output type.
|
org.tribuo.interop.tensorflow.sequence |
Provides an interface for working with TensorFlow sequence models, using Tribuo's
SequenceModel abstraction. |
Modifier and Type | Method and Description |
---|---|
TensorMap |
ImageConverter.convert(Example<?> example,
ImmutableFeatureMap featureIDMap)
Transform implicitly pads unseen values with zero.
|
TensorMap |
FeatureConverter.convert(Example<?> example,
ImmutableFeatureMap featureIDMap)
|
TensorMap |
DenseFeatureConverter.convert(Example<?> example,
ImmutableFeatureMap featureIDMap) |
TensorMap |
ImageConverter.convert(List<? extends Example<?>> examples,
ImmutableFeatureMap featureIDMap)
Transform implicitly pads unseen values with zero.
|
TensorMap |
FeatureConverter.convert(List<? extends Example<?>> example,
ImmutableFeatureMap featureIDMap)
|
TensorMap |
DenseFeatureConverter.convert(List<? extends Example<?>> examples,
ImmutableFeatureMap featureIDMap) |
TensorMap |
ImageConverter.convert(List<? extends SGDVector> vectors) |
TensorMap |
FeatureConverter.convert(List<? extends SGDVector> vectors)
|
TensorMap |
DenseFeatureConverter.convert(List<? extends SGDVector> vectors) |
TensorMap |
ImageConverter.convert(SGDVector vector) |
TensorMap |
FeatureConverter.convert(SGDVector vector)
|
TensorMap |
DenseFeatureConverter.convert(SGDVector vector) |
protected TensorMap |
TensorFlowSavedModelExternalModel.convertFeatures(SparseVector input) |
protected TensorMap |
TensorFlowFrozenExternalModel.convertFeatures(SparseVector input) |
protected TensorMap |
TensorFlowSavedModelExternalModel.convertFeaturesList(List<SparseVector> input) |
protected TensorMap |
TensorFlowFrozenExternalModel.convertFeaturesList(List<SparseVector> input) |
protected TensorMap |
TensorFlowSavedModelExternalModel.externalPrediction(TensorMap input)
Runs the session to make a prediction.
|
Modifier and Type | Method and Description |
---|---|
protected 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 TensorMap |
TensorFlowSavedModelExternalModel.externalPrediction(TensorMap input)
Runs the session to make a prediction.
|
protected org.tensorflow.Tensor |
TensorFlowFrozenExternalModel.externalPrediction(TensorMap input)
Runs the session to make a prediction.
|
Modifier and Type | Method and Description |
---|---|
TensorMap |
SequenceFeatureConverter.encode(List<? extends SequenceExample<?>> batch,
ImmutableFeatureMap featureMap)
Encodes a batch of examples as a feed dict.
|
TensorMap |
SequenceOutputConverter.encode(List<SequenceExample<T>> batch,
ImmutableOutputInfo<T> labelMap)
Encodes a batch of labels as a feed dict.
|
TensorMap |
SequenceFeatureConverter.encode(SequenceExample<?> example,
ImmutableFeatureMap featureMap)
Encodes an example as a feed dict.
|
TensorMap |
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.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.