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 TypeClassDescriptionclassCan convert aLabelinto aTensorcontaining a 32-bit integer and can convert a vector of 32-bit floats into aPredictionor aLabel.classCan convert aRegressorto aTensorcontaining a 32-bit float and a Tensor into aPredictionor 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.