Example usage for org.deeplearning4j.nn.weights WeightInit UNIFORM

Introduction

In this page you can find the example usage for org.deeplearning4j.nn.weights WeightInit UNIFORM.

Prototype

WeightInit UNIFORM

To view the source code for org.deeplearning4j.nn.weights WeightInit UNIFORM.

Click Source Link

Usage

From source file:org.wso2.carbon.ml.rest.api.neuralNetworks.FeedForwardNetwork.java

License:Open Source License

/**
 * method to map user selected WeightInit Algorithm to WeightInit object.
 * @param weightinit//from  w  ww .j  av  a 2s.c om
 * @return an WeightInit object .
 */
WeightInit mapWeightInit(String weightinit) {

    WeightInit weightInitAlgo = null;

    switch (weightinit) {

    case "Distribution":
        weightInitAlgo = WeightInit.DISTRIBUTION;
        break;

    case "Normalized":
        weightInitAlgo = WeightInit.NORMALIZED;
        break;

    case "Size":
        weightInitAlgo = WeightInit.SIZE;
        break;

    case "Uniform":
        weightInitAlgo = WeightInit.UNIFORM;
        break;

    case "Vi":
        weightInitAlgo = WeightInit.VI;
        break;

    case "Zero":
        weightInitAlgo = WeightInit.ZERO;
        break;

    case "Xavier":
        weightInitAlgo = WeightInit.XAVIER;
        break;

    case "RELU":
        weightInitAlgo = WeightInit.RELU;
        break;

    default:
        weightInitAlgo = null;
        break;
    }

    return weightInitAlgo;
}

---