Modifier and Type | Method and Description |
---|---|
MultiLayerNetwork |
LocalFileModelSaver.getBestModel() |
MultiLayerNetwork |
LocalFileModelSaver.getLatestModel() |
Modifier and Type | Method and Description |
---|---|
void |
LocalFileModelSaver.saveBestModel(MultiLayerNetwork net,
double score) |
void |
LocalFileModelSaver.saveLatestModel(MultiLayerNetwork net,
double score) |
Modifier and Type | Method and Description |
---|---|
protected org.nd4j.linalg.api.ndarray.INDArray[] |
BaseMLNScoreCalculator.output(MultiLayerNetwork network,
org.nd4j.linalg.api.ndarray.INDArray[] input,
org.nd4j.linalg.api.ndarray.INDArray[] fMask,
org.nd4j.linalg.api.ndarray.INDArray[] lMask) |
protected org.nd4j.linalg.api.ndarray.INDArray |
BaseMLNScoreCalculator.output(MultiLayerNetwork network,
org.nd4j.linalg.api.ndarray.INDArray input,
org.nd4j.linalg.api.ndarray.INDArray fMask,
org.nd4j.linalg.api.ndarray.INDArray lMask) |
protected double |
BaseMLNScoreCalculator.scoreMinibatch(MultiLayerNetwork network,
org.nd4j.linalg.api.ndarray.INDArray[] features,
org.nd4j.linalg.api.ndarray.INDArray[] labels,
org.nd4j.linalg.api.ndarray.INDArray[] fMask,
org.nd4j.linalg.api.ndarray.INDArray[] lMask,
org.nd4j.linalg.api.ndarray.INDArray[] output) |
Constructor and Description |
---|
EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train) |
EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
EarlyStoppingListener<MultiLayerNetwork> listener) |
Constructor and Description |
---|
EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> earlyStoppingConfiguration,
MultiLayerConfiguration configuration,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train) |
EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train) |
EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
EarlyStoppingListener<MultiLayerNetwork> listener) |
EarlyStoppingTrainer(EarlyStoppingConfiguration<MultiLayerNetwork> esConfig,
MultiLayerNetwork net,
org.nd4j.linalg.dataset.api.iterator.DataSetIterator train,
EarlyStoppingListener<MultiLayerNetwork> listener) |
Modifier and Type | Method and Description |
---|---|
void |
Evaluation.eval(org.nd4j.linalg.api.ndarray.INDArray trueLabels,
org.nd4j.linalg.api.ndarray.INDArray input,
MultiLayerNetwork 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(MultiLayerNetwork mln,
double epsilon,
double maxRelError,
double minAbsoluteError,
boolean print,
boolean exitOnFirstError,
org.nd4j.linalg.api.ndarray.INDArray input,
org.nd4j.linalg.api.ndarray.INDArray labels)
Check backprop gradients for a MultiLayerNetwork.
|
static boolean |
GradientCheckUtil.checkGradients(MultiLayerNetwork mln,
double epsilon,
double maxRelError,
double minAbsoluteError,
boolean print,
boolean exitOnFirstError,
org.nd4j.linalg.api.ndarray.INDArray input,
org.nd4j.linalg.api.ndarray.INDArray labels,
org.nd4j.linalg.api.ndarray.INDArray inputMask,
org.nd4j.linalg.api.ndarray.INDArray labelMask) |
static boolean |
GradientCheckUtil.checkGradients(MultiLayerNetwork mln,
double epsilon,
double maxRelError,
double minAbsoluteError,
boolean print,
boolean exitOnFirstError,
org.nd4j.linalg.api.ndarray.INDArray input,
org.nd4j.linalg.api.ndarray.INDArray labels,
org.nd4j.linalg.api.ndarray.INDArray inputMask,
org.nd4j.linalg.api.ndarray.INDArray labelMask,
boolean subset,
int maxPerParam) |
static boolean |
GradientCheckUtil.checkGradients(MultiLayerNetwork mln,
double epsilon,
double maxRelError,
double minAbsoluteError,
boolean print,
boolean exitOnFirstError,
org.nd4j.linalg.api.ndarray.INDArray input,
org.nd4j.linalg.api.ndarray.INDArray labels,
org.nd4j.linalg.api.ndarray.INDArray inputMask,
org.nd4j.linalg.api.ndarray.INDArray labelMask,
boolean subset,
int maxPerParam,
Set<String> excludeParams) |
Modifier and Type | Method and Description |
---|---|
MultiLayerNetwork |
MultiLayerNetwork.clone()
Clones the multilayernetwork
|
static MultiLayerNetwork |
MultiLayerNetwork.load(File f,
boolean loadUpdater)
Restore a MultiLayerNetwork to a file, saved using
save(File) or ModelSerializer |
Modifier and Type | Method and Description |
---|---|
void |
MultiLayerNetwork.update(MultiLayerNetwork network)
Assigns the parameters of this model to the ones specified by this
network.
|
Modifier and Type | Method and Description |
---|---|
MultiLayerNetwork |
TransferLearning.Builder.build()
Returns a model with the fine tune configuration and specified architecture changes.
|
MultiLayerNetwork |
TransferLearningHelper.unfrozenMLN()
Returns the unfrozen layers of the MultiLayerNetwork as a multilayernetwork
Note that with each call to featurizedFit the parameters to the original MLN are also updated
|
Constructor and Description |
---|
Builder(MultiLayerNetwork origModel)
Multilayer Network to tweak for transfer learning
|
TransferLearningHelper(MultiLayerNetwork orig)
Expects a MLN where some layers are frozen
|
TransferLearningHelper(MultiLayerNetwork orig,
int frozenTill)
Will modify the given MLN (in place!) to freeze layers (hold params constant during training) specified and below
|
Constructor and Description |
---|
MultiLayerUpdater(MultiLayerNetwork network) |
MultiLayerUpdater(MultiLayerNetwork network,
org.nd4j.linalg.api.ndarray.INDArray updaterState) |
Modifier and Type | Method and Description |
---|---|
MultiLayerNetwork |
CheckpointListener.loadCheckpointMLN(Checkpoint checkpoint)
Load a MultiLayerNetwork for the given checkpoint
|
MultiLayerNetwork |
CheckpointListener.loadCheckpointMLN(int checkpointNum)
Load a MultiLayerNetwork for the given checkpoint number
|
Modifier and Type | Method and Description |
---|---|
static MultiLayerNetwork |
ModelSerializer.restoreMultiLayerNetwork(File file)
Load a multi layer network from a file
|
static MultiLayerNetwork |
ModelSerializer.restoreMultiLayerNetwork(File file,
boolean loadUpdater)
Load a multi layer network from a file
|
static MultiLayerNetwork |
ModelSerializer.restoreMultiLayerNetwork(InputStream is)
Restore a multi layer network from an input stream
* Note: the input stream is read fully and closed by this method. |
static MultiLayerNetwork |
ModelSerializer.restoreMultiLayerNetwork(InputStream is,
boolean loadUpdater)
Load a MultiLayerNetwork from InputStream from an input stream
Note: the input stream is read fully and closed by this method. |
static MultiLayerNetwork |
ModelSerializer.restoreMultiLayerNetwork(String path)
Load a MultilayerNetwork model from a file
|
static MultiLayerNetwork |
ModelSerializer.restoreMultiLayerNetwork(String path,
boolean loadUpdater)
Load a MultilayerNetwork model from a file
|
Modifier and Type | Method and Description |
---|---|
static org.nd4j.linalg.primitives.Pair<MultiLayerNetwork,org.nd4j.linalg.dataset.api.preprocessor.Normalizer> |
ModelSerializer.restoreMultiLayerNetworkAndNormalizer(File file,
boolean loadUpdater)
Restore a MultiLayerNetwork and Normalizer (if present - null if not) from a File
|
static org.nd4j.linalg.primitives.Pair<MultiLayerNetwork,org.nd4j.linalg.dataset.api.preprocessor.Normalizer> |
ModelSerializer.restoreMultiLayerNetworkAndNormalizer(InputStream is,
boolean loadUpdater)
Restore a MultiLayerNetwork and Normalizer (if present - null if not) from the InputStream.
|
Modifier and Type | Method and Description |
---|---|
static void |
NetworkUtils.setLearningRate(MultiLayerNetwork net,
double newLr)
Set the learning rate for all layers in the network to the specified value.
|
static void |
NetworkUtils.setLearningRate(MultiLayerNetwork net,
int layerNumber,
double newLr)
Set the learning rate for a single layer in the network to the specified value.
|
static void |
NetworkUtils.setLearningRate(MultiLayerNetwork net,
int layerNumber,
org.nd4j.linalg.schedule.ISchedule lrSchedule)
Set the learning rate schedule for a single layer in the network to the specified value.
Note also that NetworkUtils.setLearningRate(MultiLayerNetwork, ISchedule) should also be used in preference, when all layers need
to be set to a new LR schedule.This schedule will replace any/all existing schedules, and also any fixed learning rate values. Note also that the iteration/epoch counts will not be reset. |
static void |
NetworkUtils.setLearningRate(MultiLayerNetwork net,
org.nd4j.linalg.schedule.ISchedule newLrSchedule)
Set the learning rate schedule for all layers in the network to the specified schedule.
|
static ComputationGraph |
NetworkUtils.toComputationGraph(MultiLayerNetwork net)
Convert a MultiLayerNetwork to a ComputationGraph
|
Copyright © 2018. All rights reserved.