Uses of Interface
org.tribuo.interop.tensorflow.OutputConverter
Packages that use 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
Classes in org.tribuo.interop.tensorflow that implement OutputConverterModifier and TypeClassDescriptionclassCan convert aLabelinto aTensorcontaining one hot encoding of the label and can convert aTFloat16orTFloat32into aPredictionor aLabel.classCan convert aMultiLabelinto aTensorcontaining a binary encoding of the label vector and can convert aTFloat16orTFloat32into aPredictionor aMultiLabel.classFields in org.tribuo.interop.tensorflow declared as OutputConverterModifier and TypeFieldDescriptionprotected final OutputConverter<T> TensorFlowModel.outputConverterMethods in org.tribuo.interop.tensorflow with parameters of type OutputConverterModifier and TypeMethodDescriptionstatic <T extends Output<T>>
TensorFlowFrozenExternalModel<T> TensorFlowFrozenExternalModel.createTensorflowModel(OutputFactory<T> factory, Map<String, Integer> featureMapping, Map<T, Integer> outputMapping, String inputName, 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.Constructors in org.tribuo.interop.tensorflow with parameters of type OutputConverterModifierConstructorDescriptionprotectedTensorFlowModel(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.