@Test public void testListOverloads() { MultiLayerConfiguration conf = new NeuralNetConfiguration.Builder() .seed(12345) .list() .layer(0, new DenseLayer.Builder().nIn(3).nOut(4).build()) .layer(1, new OutputLayer.Builder().nIn(4).nOut(5).build()) .pretrain(false) .backprop(true) .build(); MultiLayerNetwork net = new MultiLayerNetwork(conf); net.init(); DenseLayer dl = (DenseLayer) conf.getConf(0).getLayer(); assertEquals(3, dl.getNIn()); assertEquals(4, dl.getNOut()); OutputLayer ol = (OutputLayer) conf.getConf(1).getLayer(); assertEquals(4, ol.getNIn()); assertEquals(5, ol.getNOut()); MultiLayerConfiguration conf2 = new NeuralNetConfiguration.Builder() .seed(12345) .list() .layer(0, new DenseLayer.Builder().nIn(3).nOut(4).build()) .layer(1, new OutputLayer.Builder().nIn(4).nOut(5).build()) .pretrain(false) .backprop(true) .build(); MultiLayerNetwork net2 = new MultiLayerNetwork(conf2); net2.init(); MultiLayerConfiguration conf3 = new NeuralNetConfiguration.Builder() .seed(12345) .list( new DenseLayer.Builder().nIn(3).nOut(4).build(), new OutputLayer.Builder().nIn(4).nOut(5).build()) .pretrain(false) .backprop(true) .build(); MultiLayerNetwork net3 = new MultiLayerNetwork(conf3); net3.init(); assertEquals(conf, conf2); assertEquals(conf, conf3); }
public void printNet() { inputLayer.printLayer(inputLayer); System.out.println(); hiddenLayer.printLayer(listOfHiddenLayer); System.out.println(); outputLayer.printLayer(outputLayer); }
public void initNet() { inputLayer = new InputLayer(); inputLayer.setNumberOfNeuronsInLayer(2); numberOfHiddenLayers = 2; listOfHiddenLayer = new ArrayList<HiddenLayer>(); for (int i = 0; i < numberOfHiddenLayers; i++) { hiddenLayer = new HiddenLayer(); hiddenLayer.setNumberOfNeuronsInLayer(3); listOfHiddenLayer.add(hiddenLayer); } outputLayer = new OutputLayer(); outputLayer.setNumberOfNeuronsInLayer(1); inputLayer = inputLayer.initLayer(inputLayer); listOfHiddenLayer = hiddenLayer.initLayer(hiddenLayer, listOfHiddenLayer, inputLayer, outputLayer); outputLayer = outputLayer.initLayer(outputLayer); }