Example usage for org.deeplearning4j.nn.api OptimizationAlgorithm LINE_GRADIENT_DESCENT

Introduction

In this page you can find the example usage for org.deeplearning4j.nn.api OptimizationAlgorithm LINE_GRADIENT_DESCENT.

Prototype

OptimizationAlgorithm LINE_GRADIENT_DESCENT

To view the source code for org.deeplearning4j.nn.api OptimizationAlgorithm LINE_GRADIENT_DESCENT.

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 Optimazation Algorithm to OptimizationAlgorithm object.
 * @param optimizationAlgorithms/*  w  w w . j  a v  a  2  s .c o m*/
 * @return an OptimizationAlgorithm object.
 */

OptimizationAlgorithm mapOptimizationAlgorithm(String optimizationAlgorithms) {

    OptimizationAlgorithm optimizationAlgo = null;

    switch (optimizationAlgorithms) {
    case "Line_Gradient_Descent":
        optimizationAlgo = OptimizationAlgorithm.LINE_GRADIENT_DESCENT;
        break;

    case "Conjugate_Gradient":
        optimizationAlgo = OptimizationAlgorithm.CONJUGATE_GRADIENT;
        break;

    case "Hessian_Free":
        optimizationAlgo = OptimizationAlgorithm.HESSIAN_FREE;
        break;

    case "LBFGS":
        optimizationAlgo = OptimizationAlgorithm.LBFGS;
        break;

    case "Stochastic_Gradient_Descent":
        optimizationAlgo = OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT;
        break;

    default:
        optimizationAlgo = null;
        break;
    }

    return optimizationAlgo;
}

---