Uses of Interface
org.tribuo.interop.tensorflow.OutputConverter
Package
Description
Provides an interface to TensorFlow, allowing the training of non-sequential models using any supported
Tribuo output type.
-
Uses of OutputConverter in org.tribuo.interop.tensorflow
Modifier and TypeClassDescriptionclass
Can convert aLabel
into aTensor
containing one hot encoding of the label and can convert aTFloat16
orTFloat32
into aPrediction
or aLabel
.class
Can convert aMultiLabel
into aTensor
containing a binary encoding of the label vector and can convert aTFloat16
orTFloat32
into aPrediction
or aMultiLabel
.class
Modifier and TypeFieldDescriptionprotected final OutputConverter<T>
TensorFlowModel.outputConverter
Modifier and TypeMethodDescriptionstatic <T extends Output<T>>
TensorFlowFrozenExternalModel<T>TensorFlowFrozenExternalModel.createTensorflowModel
(OutputFactory<T> factory, Map<String, Integer> featureMapping, Map<T, Integer> outputMapping, String outputName, FeatureConverter featureConverter, OutputConverter<T> outputConverter, String filename) Creates a TensorflowFrozenExternalModel by loading in a frozen graph.static <T extends Output<T>>
TensorFlowSavedModelExternalModel<T>TensorFlowSavedModelExternalModel.createTensorflowModel
(OutputFactory<T> factory, Map<String, Integer> featureMapping, Map<T, Integer> outputMapping, String outputName, FeatureConverter featureConverter, OutputConverter<T> outputConverter, String bundleDirectory) Creates a TensorflowSavedModelExternalModel by loading in aSavedModelBundle
.ModifierConstructorDescriptionprotected
TensorFlowModel
(String name, ModelProvenance provenance, ImmutableFeatureMap featureIDMap, ImmutableOutputInfo<T> outputIDInfo, org.tensorflow.proto.framework.GraphDef trainedGraphDef, int batchSize, String outputName, FeatureConverter featureConverter, OutputConverter<T> outputConverter) Builds a TFModel.TensorFlowTrainer
(Path graphPath, String outputName, GradientOptimiser optimiser, Map<String, Float> gradientParams, FeatureConverter featureConverter, OutputConverter<T> outputConverter, int trainBatchSize, int epochs, int testBatchSize, int loggingInterval) Constructs a Trainer for a TensorFlow graph.TensorFlowTrainer
(Path graphPath, String outputName, GradientOptimiser optimiser, Map<String, Float> gradientParams, FeatureConverter featureConverter, OutputConverter<T> outputConverter, int trainBatchSize, int epochs, int testBatchSize, int loggingInterval, Path checkpointPath) Constructs a Trainer for a TensorFlow graph.TensorFlowTrainer
(org.tensorflow.Graph graph, String outputName, GradientOptimiser optimiser, Map<String, Float> gradientParams, FeatureConverter featureConverter, OutputConverter<T> outputConverter, int trainBatchSize, int epochs, int testBatchSize, int loggingInterval) Constructs a Trainer for a TensorFlow graph.TensorFlowTrainer
(org.tensorflow.Graph graph, String outputName, GradientOptimiser optimiser, Map<String, Float> gradientParams, FeatureConverter featureConverter, OutputConverter<T> outputConverter, int trainBatchSize, int epochs, int testBatchSize, int loggingInterval, Path checkpointPath) Constructs a Trainer for a TensorFlow graph.TensorFlowTrainer
(org.tensorflow.proto.framework.GraphDef graphDef, String outputName, GradientOptimiser optimiser, Map<String, Float> gradientParams, FeatureConverter featureConverter, OutputConverter<T> outputConverter, int trainBatchSize, int epochs, int testBatchSize, int loggingInterval) Constructs a Trainer for a TensorFlow graph.TensorFlowTrainer
(org.tensorflow.proto.framework.GraphDef graphDef, String outputName, GradientOptimiser optimiser, Map<String, Float> gradientParams, FeatureConverter featureConverter, OutputConverter<T> outputConverter, int trainBatchSize, int epochs, int testBatchSize, int loggingInterval, Path checkpointPath) Constructs a Trainer for a TensorFlow graph.