|
Java example source code file (Dropout.java)
The Dropout.java Java example source codepackage org.deeplearning4j.util; import org.deeplearning4j.nn.api.Layer; import org.nd4j.linalg.api.ndarray.INDArray; import org.nd4j.linalg.api.ops.impl.transforms.DropOutInverted; import org.nd4j.linalg.api.shape.Shape; import org.nd4j.linalg.factory.Nd4j; /** * @author Adam Gibson */ public class Dropout { /** * Apply drop connect to the given variable * @param layer the layer with the variables * @param variable the variable to apply * @return the post applied drop connect */ public static INDArray applyDropConnect(Layer layer,String variable) { return layer.getParam(variable).mul(Nd4j.getDistributions().createBinomial(1,layer.conf().getLayer().getDropOut()).sample(layer.getParam(variable).shape())); } /** * Apply dropout to the given input * and return the drop out mask used * @param input the input to do drop out on * @param dropout the drop out probability */ public static void applyDropout(INDArray input,double dropout) { Nd4j.getExecutioner().exec(new DropOutInverted(input, dropout)); } } Other Java examples (source code examples)Here is a short list of links related to this Java Dropout.java source code file: |
... this post is sponsored by my books ... | |
#1 New Release! |
FP Best Seller |
Copyright 1998-2021 Alvin Alexander, alvinalexander.com
All Rights Reserved.
A percentage of advertising revenue from
pages under the /java/jwarehouse
URI on this website is
paid back to open source projects.