public static final class Dropout.Builder
extends java.lang.Object
Modifier and Type | Method and Description |
---|---|
Dropout |
build()
Builds a
Dropout block. |
Dropout.Builder |
optProbability(float probability)
Sets the probability or the fraction of the input that gets dropped out during training
time.
|
Dropout.Builder |
optSharedAxes(int[] sharedAxes)
Sets the axes for variational dropout kernel.
|
public Dropout.Builder optProbability(float probability)
probability
- fraction of the input that gets dropped out during trainingpublic Dropout.Builder optSharedAxes(int[] sharedAxes)
sharedAxes
- the axes for variational dropout kernel