Ejemplo n.º 1
0
    def runTest(self):
        nn = MLP(*self.args)
        for layer in nn.layers:
            layer.b = util.uniform(layer.b.shape, 0.5)
        X_train = self.X_train
        y_train = self.y_train

        loss_before = nn.loss(X_train, y_train)
        nn.dump('test.npz')

        nn = MLP(*self.args)
        nn.load('test.npz')
        loss_after = nn.loss(X_train, y_train)

        np.testing.assert_almost_equal(loss_after, loss_before)
        print loss_before, loss_after