Package | Description |
---|---|
org.deeplearning4j.nn | |
org.deeplearning4j.nn.conf |
Modifier and Type | Method and Description |
---|---|
static WeightInit |
WeightInit.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static WeightInit[] |
WeightInit.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Modifier and Type | Method and Description |
---|---|
static org.nd4j.linalg.api.ndarray.INDArray |
WeightInitUtil.initWeights(int nIn,
int nOut,
WeightInit initScheme,
org.nd4j.linalg.api.activation.ActivationFunction act,
org.apache.commons.math3.distribution.RealDistribution dist)
Initializes a matrix with the given weight initialization scheme
|
Modifier and Type | Field and Description |
---|---|
protected WeightInit |
NeuralNetConfiguration.weightInit |
Modifier and Type | Method and Description |
---|---|
WeightInit |
NeuralNetConfiguration.getWeightInit() |
Modifier and Type | Method and Description |
---|---|
void |
NeuralNetConfiguration.setWeightInit(WeightInit weightInit) |
NeuralNetConfiguration.Builder |
NeuralNetConfiguration.Builder.weightInit(WeightInit weightInit) |
Constructor and Description |
---|
NeuralNetConfiguration(float sparsity,
boolean useAdaGrad,
float lr,
int k,
float corruptionLevel,
int numIterations,
float momentum,
float l2,
boolean useRegularization,
Map<Integer,Float> momentumAfter,
int resetAdaGradIterations,
float dropOut,
boolean applySparsity,
WeightInit weightInit,
NeuralNetwork.OptimizationAlgorithm optimizationAlgo,
org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction lossFunction,
int renderWeightsEveryNumEpochs,
boolean concatBiases,
boolean constrainGradientToUnitNorm,
org.apache.commons.math3.random.RandomGenerator rng,
org.apache.commons.math3.distribution.RealDistribution dist,
long seed,
int nIn,
int nOut,
org.nd4j.linalg.api.activation.ActivationFunction activationFunction,
RBM.VisibleUnit visibleUnit,
RBM.HiddenUnit hiddenUnit,
NeuralNetConfiguration.ActivationType activationType,
int[] weightShape,
int[] filterSize,
int numFeatureMaps,
int[] stride,
int[] featureMapSize,
int numInFeatureMaps) |
Copyright © 2014. All rights reserved.