@Test
public void testBuilder() {
DianneNeuralNetworkBuilder builder = new DianneNeuralNetworkBuilder("Test");
builder.addLinear(784, 20);
builder.addReLU();
builder.addLinear(20, 10);
builder.addSoftmax();
NeuralNetworkDTO nn = builder.create();
System.out.println(DianneJSONConverter.toJsonString(nn, true));
}
@Test
public void testMLP() {
NeuralNetworkDTO nn =
DianneNeuralNetworkBuilder.createMLP("MLP", 784, 10, Activation.Sigmoid, 100, 20);
System.out.println(DianneJSONConverter.toJsonString(nn, true));
}
