Package org.deeplearning4j.nn.weights
Enum WeightInit
- java.lang.Object
-
- java.lang.Enum<WeightInit>
-
- org.deeplearning4j.nn.weights.WeightInit
-
- All Implemented Interfaces:
Serializable
,Comparable<WeightInit>
public enum WeightInit extends Enum<WeightInit>
-
-
Enum Constant Summary
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description IWeightInit
getWeightInitFunction()
Create an instance of the weight initialization functionIWeightInit
getWeightInitFunction(Distribution distribution)
Create an instance of the weight initialization functionstatic WeightInit
valueOf(String name)
Returns the enum constant of this type with the specified name.static WeightInit[]
values()
Returns an array containing the constants of this enum type, in the order they are declared.
-
-
-
Enum Constant Detail
-
DISTRIBUTION
public static final WeightInit DISTRIBUTION
-
ZERO
public static final WeightInit ZERO
-
ONES
public static final WeightInit ONES
-
SIGMOID_UNIFORM
public static final WeightInit SIGMOID_UNIFORM
-
NORMAL
public static final WeightInit NORMAL
-
LECUN_NORMAL
public static final WeightInit LECUN_NORMAL
-
UNIFORM
public static final WeightInit UNIFORM
-
XAVIER
public static final WeightInit XAVIER
-
XAVIER_UNIFORM
public static final WeightInit XAVIER_UNIFORM
-
XAVIER_FAN_IN
public static final WeightInit XAVIER_FAN_IN
-
XAVIER_LEGACY
public static final WeightInit XAVIER_LEGACY
-
RELU
public static final WeightInit RELU
-
RELU_UNIFORM
public static final WeightInit RELU_UNIFORM
-
IDENTITY
public static final WeightInit IDENTITY
-
LECUN_UNIFORM
public static final WeightInit LECUN_UNIFORM
-
VAR_SCALING_NORMAL_FAN_IN
public static final WeightInit VAR_SCALING_NORMAL_FAN_IN
-
VAR_SCALING_NORMAL_FAN_OUT
public static final WeightInit VAR_SCALING_NORMAL_FAN_OUT
-
VAR_SCALING_NORMAL_FAN_AVG
public static final WeightInit VAR_SCALING_NORMAL_FAN_AVG
-
VAR_SCALING_UNIFORM_FAN_IN
public static final WeightInit VAR_SCALING_UNIFORM_FAN_IN
-
VAR_SCALING_UNIFORM_FAN_OUT
public static final WeightInit VAR_SCALING_UNIFORM_FAN_OUT
-
VAR_SCALING_UNIFORM_FAN_AVG
public static final WeightInit VAR_SCALING_UNIFORM_FAN_AVG
-
-
Method Detail
-
values
public static WeightInit[] values()
Returns an array containing the constants of this enum type, in the order they are declared. This method may be used to iterate over the constants as follows:for (WeightInit c : WeightInit.values()) System.out.println(c);
- Returns:
- an array containing the constants of this enum type, in the order they are declared
-
valueOf
public static WeightInit valueOf(String name)
Returns the enum constant of this type with the specified name. The string must match exactly an identifier used to declare an enum constant in this type. (Extraneous whitespace characters are not permitted.)- Parameters:
name
- the name of the enum constant to be returned.- Returns:
- the enum constant with the specified name
- Throws:
IllegalArgumentException
- if this enum type has no constant with the specified nameNullPointerException
- if the argument is null
-
getWeightInitFunction
public IWeightInit getWeightInitFunction()
Create an instance of the weight initialization function- Returns:
- a new
IWeightInit
instance
-
getWeightInitFunction
public IWeightInit getWeightInitFunction(Distribution distribution)
Create an instance of the weight initialization function- Parameters:
distribution
- Distribution of the weights (Only used in case DISTRIBUTION)- Returns:
- a new
IWeightInit
instance
-
-