org.tribuo.interop.tensorflow

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

Fields

Modifier and Type	Field and Description
int	batchSize
Path	checkpointPath
int	epochs
List<String>	gradientParamNames
List<Float>	gradientParamValues
String	imageFormat
String	inputName
TrainTest.InputType	inputType
int	loggingInterval
GradientOptimiser	optimiser
String	outputName
Path	outputPath
Path	protobufPath
int	testBatchSize
Path	testingPath
Path	trainingPath

Fields inherited from interface com.oracle.labs.mlrg.olcut.config.Options

header

Constructor Summary

Constructors

Constructor and Description
TensorflowOptions()

Method Summary

Modifier and Type	Method and Description
Map<String,Float>	getGradientParams() Zips the gradient parameter names and values.
String	getOptionsDescription()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface com.oracle.labs.mlrg.olcut.config.Options

formatUsage, generateTypeDescription, getAllOptions, getEnumConstantString, getOptionFields, getOptions, getOptionUsage, getOptionUsage, getOptionUsage, getUsage

Field Detail

outputPath

@Option(charName=102,
        longName="model-output-path",
        usage="Path to serialize model to.")
public Path outputPath

trainingPath

@Option(charName=117,
        longName="training-file",
        usage="Path to the libsvm format training file.")
public Path trainingPath

testingPath

@Option(charName=118,
        longName="testing-file",
        usage="Path to the libsvm format testing file.")
public Path testingPath

outputName

@Option(charName=108,
        longName="output-name",
        usage="Name of the output operation.")
public String outputName

gradientParamNames

@Option(longName="optimizer-param-names",
        usage="Gradient optimizer param names, see org.tribuo.interop.tensorflow.GradientOptimiser.")
public List<String> gradientParamNames

gradientParamValues

@Option(longName="optimizer-param-values",
        usage="Gradient optimizer param values, see org.tribuo.interop.tensorflow.GradientOptimiser.")
public List<Float> gradientParamValues

optimiser

@Option(charName=103,
        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=98,
        longName="batch-size",
        usage="Minibatch size.")
public int batchSize

epochs

@Option(charName=101,
        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=110,
        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=116,
        longName="input-type",
        usage="Input type.")
public TrainTest.InputType inputType

protobufPath

@Option(charName=109,
        longName="model-protobuf",
        usage="Path to the protobuf containing the network description.")
public Path protobufPath

checkpointPath

@Option(charName=112,
        longName="checkpoint-dir",
        usage="Path to the checkpoint base directory.")
public Path checkpointPath

Constructor Detail

TensorflowOptions

public TensorflowOptions()

Method Detail

getOptionsDescription

public String getOptionsDescription()

Specified by:

getOptionsDescription in interface com.oracle.labs.mlrg.olcut.config.Options

getGradientParams

public Map<String,Float> getGradientParams()

Zips the gradient parameter names and values.

Returns:

The gradient parameter map.