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.
-
ClassDescriptionSequenceExampleTransformer<T extends Output<T>>Converts a sequence example into a feed dict suitable for Tensorflow.SequenceOutputTransformer<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.