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)
还没有任何评论哟~
