org.deeplearning4j.util

Class Dropout

java.lang.Object

org.deeplearning4j.util.Dropout

public class Dropout
extends Object

Author:

Adam Gibson

Method Summary

Modifier and Type	Method and Description
static org.nd4j.linalg.api.ndarray.INDArray	applyDropConnect(Layer layer, String variable) Apply drop connect to the given variable
static void	applyDropout(org.nd4j.linalg.api.ndarray.INDArray input, double dropout) Apply dropout to the given input and return the drop out mask used

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

applyDropConnect

public static org.nd4j.linalg.api.ndarray.INDArray applyDropConnect(Layer layer,
                                                                    String variable)

Apply drop connect to the given variable

Parameters:

layer - the layer with the variables

variable - the variable to apply

Returns:

the post applied drop connect

applyDropout

public static void applyDropout(org.nd4j.linalg.api.ndarray.INDArray input,
                                double dropout)

Apply dropout to the given input and return the drop out mask used

Parameters:

input - the input to do drop out on

dropout - the drop out probability