Uses of Enum Class
org.tribuo.interop.tensorflow.GradientOptimiser
Package
Description
Provides an interface to TensorFlow, allowing the training of non-sequential models using any supported
Tribuo output type.
-
Uses of GradientOptimiser in org.tribuo.interop.tensorflow
Modifier and TypeFieldDescriptionTrainTest.TensorflowOptions.optimiser
The gradient optimizer to use.Modifier and TypeMethodDescriptionstatic GradientOptimiser
Returns the enum constant of this class with the specified name.static GradientOptimiser[]
GradientOptimiser.values()
Returns an array containing the constants of this enum class, in the order they are declared.ModifierConstructorDescriptionTensorFlowTrainer
(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.