Uses of Interface
org.tribuo.interop.tensorflow.OutputTransformer
Packages that use OutputTransformer
Package
Description
Provides an interface to Tensorflow, allowing the training of non-sequential models using any supported
Tribuo output type.
-
Uses of OutputTransformer in org.tribuo.interop.tensorflow
Classes in org.tribuo.interop.tensorflow that implement OutputTransformerModifier and TypeClassDescriptionclass
Can convert aLabel
into aTensor
containing a 32-bit integer and can convert a vector of 32-bit floats into aPrediction
or aLabel
.class
Can convert aRegressor
to aTensor
containing a 32-bit float and a Tensor into aPrediction
or Regressor.Methods in org.tribuo.interop.tensorflow with parameters of type OutputTransformerModifier and TypeMethodDescriptionstatic <T extends Output<T>>
TensorflowExternalModel<T> TensorflowExternalModel.createTensorflowModel
(OutputFactory<T> factory, Map<String, Integer> featureMapping, Map<T, Integer> outputMapping, String inputName, String outputName, ExampleTransformer<T> featureTransformer, OutputTransformer<T> outputTransformer, String filename) Creates a TensorflowExternalModel by loading in a frozen graph.Constructors in org.tribuo.interop.tensorflow with parameters of type OutputTransformerModifierConstructorDescriptionTensorflowCheckpointTrainer
(Path graphPath, Path checkpointRootPath, ExampleTransformer<T> exampleTransformer, OutputTransformer<T> outputTransformer, int minibatchSize, int epochs) Builds a trainer using the supplied graph and arguments.TensorflowTrainer
(byte[] graphDef, ExampleTransformer<T> exampleTransformer, OutputTransformer<T> outputTransformer, int minibatchSize, int epochs, int testBatchSize) Constructs a Trainer for a tensorflow graph.TensorflowTrainer
(Path graphPath, ExampleTransformer<T> exampleTransformer, OutputTransformer<T> outputTransformer, int minibatchSize, int epochs, int testBatchSize) Constructs a Trainer for a tensorflow graph.