| Modifier and Type | Method and Description |
|---|---|
ComputationGraph |
LocalFileGraphSaver.getBestModel() |
ComputationGraph |
LocalFileGraphSaver.getLatestModel() |
| Modifier and Type | Method and Description |
|---|---|
void |
LocalFileGraphSaver.saveBestModel(ComputationGraph net,
double score) |
void |
LocalFileGraphSaver.saveLatestModel(ComputationGraph net,
double score) |
| Modifier and Type | Method and Description |
|---|---|
double |
DataSetLossCalculatorCG.calculateScore(ComputationGraph network) |
| Constructor and Description |
|---|
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train) |
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
EarlyStoppingListener<ComputationGraph> listener)
Constructor for training using a
DataSetIterator |
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator train,
EarlyStoppingListener<ComputationGraph> listener)
Constructor for training using a
MultiDataSetIterator |
| Constructor and Description |
|---|
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train) |
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
EarlyStoppingListener<ComputationGraph> listener)
Constructor for training using a
DataSetIterator |
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
EarlyStoppingListener<ComputationGraph> listener)
Constructor for training using a
DataSetIterator |
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator train,
EarlyStoppingListener<ComputationGraph> listener)
Constructor for training using a
MultiDataSetIterator |
EarlyStoppingGraphTrainer(EarlyStoppingConfiguration<ComputationGraph> esConfig,
ComputationGraph net,
org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator train,
EarlyStoppingListener<ComputationGraph> listener)
Constructor for training using a
MultiDataSetIterator |
| Modifier and Type | Method and Description |
|---|---|
void |
Evaluation.eval(org.nd4j.linalg.api.ndarray.INDArray trueLabels,
org.nd4j.linalg.api.ndarray.INDArray input,
ComputationGraph network)
Evaluate the output
using the given true labels,
the input to the multi layer network
and the multi layer network to
use for evaluation
|
| Modifier and Type | Method and Description |
|---|---|
static boolean |
GradientCheckUtil.checkGradients(ComputationGraph graph,
double epsilon,
double maxRelError,
double minAbsoluteError,
boolean print,
boolean exitOnFirstError,
org.nd4j.linalg.api.ndarray.INDArray[] inputs,
org.nd4j.linalg.api.ndarray.INDArray[] labels)
Check backprop gradients for a ComputationGraph
|
| Modifier and Type | Method and Description |
|---|---|
GraphVertex |
ScaleVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
GraphVertex |
PreprocessorVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
GraphVertex |
PoolHelperVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
GraphVertex |
LayerVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
GraphVertex |
MergeVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
GraphVertex |
L2NormalizeVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
GraphVertex |
SubsetVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
abstract GraphVertex |
GraphVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams)
Create a
GraphVertex instance, for the given computation graph,
given the configuration instance. |
GraphVertex |
ShiftVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
GraphVertex |
ElementWiseVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
GraphVertex |
UnstackVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
GraphVertex |
StackVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
GraphVertex |
L2Vertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
GraphVertex |
ReshapeVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
| Modifier and Type | Method and Description |
|---|---|
GraphVertex |
DuplicateToTimeSeriesVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
LastTimeStepVertex |
LastTimeStepVertex.instantiate(ComputationGraph graph,
String name,
int idx,
org.nd4j.linalg.api.ndarray.INDArray paramsView,
boolean initializeParams) |
| Modifier and Type | Method and Description |
|---|---|
ComputationGraph |
ComputationGraph.clone() |
| Modifier and Type | Field and Description |
|---|---|
protected ComputationGraph |
BaseGraphVertex.graph |
| Constructor and Description |
|---|
BaseGraphVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices) |
| Constructor and Description |
|---|
ElementWiseVertex(ComputationGraph graph,
String name,
int vertexIndex,
ElementWiseVertex.Op op) |
ElementWiseVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices,
ElementWiseVertex.Op op) |
InputVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] outputVertices) |
L2NormalizeVertex(ComputationGraph graph,
String name,
int vertexIndex,
int[] dimension,
double eps) |
L2NormalizeVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices,
int[] dimension,
double eps) |
L2Vertex(ComputationGraph graph,
String name,
int vertexIndex,
double eps) |
L2Vertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices,
double eps) |
LayerVertex(ComputationGraph graph,
String name,
int vertexIndex,
Layer layer,
InputPreProcessor layerPreProcessor,
boolean outputVertex)
Create a network input vertex:
|
LayerVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices,
Layer layer,
InputPreProcessor layerPreProcessor,
boolean outputVertex) |
MergeVertex(ComputationGraph graph,
String name,
int vertexIndex) |
MergeVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices) |
PoolHelperVertex(ComputationGraph graph,
String name,
int vertexIndex) |
PoolHelperVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices) |
PreprocessorVertex(ComputationGraph graph,
String name,
int vertexIndex,
InputPreProcessor preProcessor) |
PreprocessorVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices,
InputPreProcessor preProcessor) |
ReshapeVertex(ComputationGraph graph,
String name,
int vertexIndex,
int[] newShape) |
ReshapeVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices,
int[] newShape) |
ScaleVertex(ComputationGraph graph,
String name,
int vertexIndex,
double scaleFactor) |
ScaleVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices,
double scaleFactor) |
ShiftVertex(ComputationGraph graph,
String name,
int vertexIndex,
double shiftFactor) |
ShiftVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices,
double shiftFactor) |
StackVertex(ComputationGraph graph,
String name,
int vertexIndex) |
StackVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices) |
SubsetVertex(ComputationGraph graph,
String name,
int vertexIndex,
int from,
int to) |
SubsetVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices,
int from,
int to) |
UnstackVertex(ComputationGraph graph,
String name,
int vertexIndex,
int from,
int stackSize) |
UnstackVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices,
int from,
int stackSize) |
| Constructor and Description |
|---|
DuplicateToTimeSeriesVertex(ComputationGraph graph,
String name,
int vertexIndex,
String inputVertexName) |
DuplicateToTimeSeriesVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices,
String inputName) |
LastTimeStepVertex(ComputationGraph graph,
String name,
int vertexIndex,
String inputName) |
LastTimeStepVertex(ComputationGraph graph,
String name,
int vertexIndex,
VertexIndices[] inputVertices,
VertexIndices[] outputVertices,
String inputName) |
| Modifier and Type | Method and Description |
|---|---|
ComputationGraph |
TransferLearning.GraphBuilder.build()
Returns a computation graph build to specifications.
|
ComputationGraph |
TransferLearningHelper.unfrozenGraph()
Returns the unfrozen subset of the original computation graph as a computation graph
Note that with each call to featurizedFit the parameters to the original computation graph are also updated
|
| Constructor and Description |
|---|
GraphBuilder(ComputationGraph origGraph)
Computation Graph to tweak for transfer learning
|
TransferLearningHelper(ComputationGraph orig)
Expects a computation graph where some vertices are frozen
|
TransferLearningHelper(ComputationGraph orig,
String... frozenOutputAt)
Will modify the given comp graph (in place!) to freeze vertices from input to the vertex specified.
|
| Constructor and Description |
|---|
ComputationGraphUpdater(ComputationGraph graph) |
ComputationGraphUpdater(ComputationGraph graph,
org.nd4j.linalg.api.ndarray.INDArray updaterState) |
| Modifier and Type | Method and Description |
|---|---|
static ComputationGraph |
ModelSerializer.restoreComputationGraph(File file)
Load a computation graph from a file
|
static ComputationGraph |
ModelSerializer.restoreComputationGraph(File file,
boolean loadUpdater)
Load a computation graph from a file
|
static ComputationGraph |
ModelSerializer.restoreComputationGraph(InputStream is)
Load a computation graph from a InputStream
|
static ComputationGraph |
ModelSerializer.restoreComputationGraph(InputStream is,
boolean loadUpdater)
Load a computation graph from a InputStream
|
static ComputationGraph |
ModelSerializer.restoreComputationGraph(String path)
Load a computation graph from a file
|
static ComputationGraph |
ModelSerializer.restoreComputationGraph(String path,
boolean loadUpdater)
Load a computation graph from a file
|
Copyright © 2017. All rights reserved.