Class TrainTest

java.lang.Object

org.tribuo.interop.tensorflow.TrainTest

public class TrainTest
extends Object

Build and run a Tensorflow multi-class classifier for a standard dataset.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static enum	TrainTest.InputType
static class	TrainTest.TensorflowOptions

Constructor Summary

Constructors

Constructor	Description
TrainTest()

Method Summary

Modifier and Type	Method	Description
static com.oracle.labs.mlrg.olcut.util.Pair<Dataset<Label>, Dataset<Label>>	load(Path trainingPath, Path testingPath, OutputFactory<Label> outputFactory)
static void	main(String[] args)
static void	saveModel(Path outputPath, Model<Label> model)

Methods inherited from class java.lang.Object

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

Constructor Details

TrainTest

public TrainTest()

Method Details

load

public static com.oracle.labs.mlrg.olcut.util.Pair<Dataset<Label>, Dataset<Label>> load(Path trainingPath,
 Path testingPath,
 OutputFactory<Label> outputFactory)
                                                                                 throws IOException

Throws:

IOException

saveModel

public static void saveModel(Path outputPath,
 Model<Label> model)
                      throws IOException

Throws:

IOException

main

public static void main(String[] args)
                 throws IOException

Parameters:

args - the command line arguments

Throws:

IOException - if there is any error reading the examples.