Package | Description |
---|---|
org.tribuo.interop.tensorflow.example |
Example architectures for use with Tribuo's TF interface.
|
Modifier and Type | Method and Description |
---|---|
static 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.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.