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

Introduction

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

Prototype

OptimizationAlgorithm LBFGS

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

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//from www.  j  av  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;
}

---