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. |
Class and Description |
---|
FeatureConverter |
GradientOptimiser
An enum for the gradient optimisers exposed by TensorFlow-Java.
|
OutputConverter
Converts the
Output into a Tensor and vice versa. |
TensorFlowCheckpointModel
This model encapsulates a simple model with an input feed dict,
and produces a single output tensor.
|
TensorFlowFrozenExternalModel
A Tribuo wrapper around a TensorFlow frozen model.
|
TensorFlowModel
Base class for a TensorFlow model that operates on
Example s. |
TensorFlowNativeModel
This model encapsulates a TensorFlow model running in graph mode with a single tensor output.
|
TensorFlowSavedModelExternalModel
A Tribuo wrapper around a TensorFlow saved model bundle.
|
TensorFlowTrainer.TFModelFormat
The model format to emit.
|
TensorFlowUtil.TensorTuple
A serializable tuple containing the tensor class name, the shape and the data.
|
TensorMap
A map of names and tensors to feed into a session.
|
TrainTest.InputType
Type of feature extractor.
|
Class and Description |
---|
TensorMap
A map of names and tensors to feed into a session.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.