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.