Package | Description |
---|---|
org.tribuo.interop.tensorflow |
Provides an interface to TensorFlow, allowing the training of non-sequential models using any supported
Tribuo output type.
|
Modifier and Type | Field and Description |
---|---|
GradientOptimiser |
TrainTest.TensorflowOptions.optimiser |
Modifier and Type | Method and Description |
---|---|
static GradientOptimiser |
GradientOptimiser.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static GradientOptimiser[] |
GradientOptimiser.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Constructor and Description |
---|
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.
|
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(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.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.