Package org.tribuo.interop.tensorflow.sequence
package org.tribuo.interop.tensorflow.sequence
Provides an interface for working with TensorFlow sequence models, using Tribuo's
SequenceModel
abstraction.
N.B. TensorFlow support is experimental and may change without a major version bump.
-
ClassDescriptionConverts a sequence example into a feed dict suitable for TensorFlow.SequenceOutputConverter<T extends Output<T>>Converts a TensorFlow output tensor into a list of predictions, and a Tribuo sequence example into a Tensorflow tensor suitable for training.TensorFlowSequenceModel<T extends Output<T>>A TensorFlow model which implements SequenceModel, suitable for use in sequential prediction tasks.TensorFlowSequenceTrainer<T extends Output<T>>A trainer for SequenceModels which use an underlying TensorFlow graph.Provenance for
TensorFlowSequenceTrainer
.