@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);
}
