Advertisement

Dl4j-fit(DataSetIterator iterator)代码阅读(一)

阅读量:

fit(DataSetIterator iterator)源码阅读

网络模型构建与分析

复制代码
    //Create the network
    int numInput = 1;
    int numOutputs = 1;
    int nHidden = 2;
    MultiLayerNetwork net = new MultiLayerNetwork(new NeuralNetConfiguration.Builder()
        .seed(seed)
        .iterations(iterations)
        .optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT)
        .learningRate(learningRate)
        .weightInit(WeightInit.XAVIER)
        .updater(Updater.SGD)     //To configure: .updater(new Nesterovs(0.9))
        .list()
        .layer(0, new DenseLayer.Builder().nIn(numInput).nOu

全部评论 (0)

还没有任何评论哟~