org.deeplearning4j.util

Class Dropout

java.lang.Object

org.deeplearning4j.util.Dropout

public class Dropout
extends Object

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
Dropout()

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 org.nd4j.linalg.api.ndarray.INDArray	applyDropout(org.nd4j.linalg.api.ndarray.INDArray input, double dropout, org.nd4j.linalg.api.ndarray.INDArray dropoutMask) 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

Constructor Detail

Dropout

public Dropout()

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 org.nd4j.linalg.api.ndarray.INDArray applyDropout(org.nd4j.linalg.api.ndarray.INDArray input,
                                                                double dropout,
                                                                org.nd4j.linalg.api.ndarray.INDArray dropoutMask)

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

dropoutMask - the dropout mask applied (can be null)

Returns:

the dropout mask used