org.tribuo.interop.tensorflow

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 and Description
static class	TrainTest.InputType
static class	TrainTest.TensorflowOptions

Constructor Summary

Constructors

Constructor and Description
TrainTest()

Method Summary

Modifier and Type	Method and 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 Detail

TrainTest

public TrainTest()

Method Detail

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.