Uses of Class
org.tribuo.interop.tensorflow.example.GraphDefTuple
Package
Description
Example architectures for use with Tribuo's TF interface.
-
Uses of GraphDefTuple in org.tribuo.interop.tensorflow.example
Modifier and TypeMethodDescriptionstatic GraphDefTuple
CNNExamples.buildLeNetGraph
(String inputName, int imageSize, int pixelDepth, int numOutputs) Builds a LeNet 5 style CNN (usually used for MNIST).static GraphDefTuple
MLPExamples.buildMLPGraph
(String inputName, int numFeatures, int[] hiddenSizes, int numOutputs) Builds an MLP which expects the supplied number of inputs, has hiddenSizes.length hidden layers, before emitting numOutput outputs.