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

  • 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

    • testBatchSize

      @Option(charName='b', longName="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

    • imageFormat

      @Option(charName='i', 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

  • Method Details

    • getOptionsDescription

      public String getOptionsDescription()
      Specified by:
      getOptionsDescription in interface com.oracle.labs.mlrg.olcut.config.Options