Class TrainTest.TensorflowOptions
java.lang.Object
org.tribuo.interop.tensorflow.TrainTest.TensorflowOptions
- All Implemented Interfaces:
com.oracle.labs.mlrg.olcut.config.Options
- Enclosing class:
TrainTest
public static class TrainTest.TensorflowOptions
extends Object
implements com.oracle.labs.mlrg.olcut.config.Options
Options for training a model in TensorFlow.
-
Field Summary
FieldsModifier and TypeFieldDescriptionint
int
int
int
Fields inherited from interface com.oracle.labs.mlrg.olcut.config.Options
header
-
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescriptionZips the gradient parameter names and values.
-
Field Details
-
outputPath
@Option(charName='f', longName="model-output-path", usage="Path to serialize model to.") public Path outputPath -
trainingPath
@Option(charName='u', longName="training-file", usage="Path to the libsvm format training file.") public Path trainingPath -
testingPath
@Option(charName='v', longName="testing-file", usage="Path to the libsvm format testing file.") public Path testingPath -
outputName
@Option(charName='l', longName="output-name", usage="Name of the output operation.") public String outputName -
gradientParamNames
-
gradientParamValues
-
optimiser
@Option(charName='g', longName="gradient-optimizer", usage="The gradient optimizer to use.") public GradientOptimiser optimiser -
testBatchSize
@Option(longName="test-batch-size", usage="Test time minibatch size.") public int testBatchSize -
batchSize
@Option(charName='b', longName="batch-size", usage="Minibatch size.") public int batchSize -
epochs
@Option(charName='e', longName="num-epochs", usage="Number of gradient descent epochs.") public int epochs -
loggingInterval
@Option(longName="logging-interval", usage="Interval between logging the loss.") public int loggingInterval -
inputName
@Option(charName='n', longName="input-name", usage="Name of the input placeholder.") public String inputName -
imageFormat
@Option(longName="image-format", usage="Image format, in [W,H,C]. Defaults to MNIST.") public String imageFormat -
inputType
@Option(charName='t', longName="input-type", usage="Input type.") public TrainTest.InputType inputType -
protobufPath
@Option(charName='m', longName="model-protobuf", usage="Path to the protobuf containing the network description.") public Path protobufPath -
checkpointPath
@Option(charName='p', longName="checkpoint-dir", usage="Path to the checkpoint base directory.") public Path checkpointPath
-
-
Constructor Details
-
TensorflowOptions
public TensorflowOptions()
-
-
Method Details
-
getOptionsDescription
- Specified by:
getOptionsDescription
in interfacecom.oracle.labs.mlrg.olcut.config.Options
-
getGradientParams
-