def test_single_layer_network(self): network = Network() network.add_layer(Normalize()) indata = np.array([2, 7, 3]) outdata = network.compute(indata) self.assertTrue(np.allclose(outdata, [-0.4, 0.6, -0.2]))
def test_multi_layer_network(self): network = Network() network.add_layer(Convolve([[0.1, 0], [0, 0.1]], keepdims=False)) network.add_layer(Relu()) network.add_layer(Flatten()) network.add_layer(Softmax()) indata = np.array([[5, 2, 7, 10], [1, 8, 3, 4]]) outdata = network.compute(indata) self.assertTrue(np.allclose(outdata, [0.440905, 0.19811, 0.36098]))