Ejemplo n.º 1
0
def mnist_net():
    mndata = MNIST("mnist", return_type="numpy")
    print("Loading images...")
    images, labels = mndata.load_training()
    features = images.T / 255
    z = np.zeros((60000, 10))
    z[np.arange(60000), labels] = 1
    Y = z.T
    nn = NN([784, 100, 30, 10])
    nn.set_hyperparameters(learning_rate=0.5)
    t = time()
    nn.initialize_parameters()
    print("Start Training...")
    nn.minimize({"features": features, "labels": Y}, 20)
    print("Finish Training.")
    print("Training time: {0} seconds".format(round(time() - t, 2)))
    print("Start Testing...")
    t = time()
    test_images, test_labels = mndata.load_testing()
    test_features = test_images.T / 255
    z = np.zeros((10000, 10))
    z[np.arange(10000), test_labels] = 1
    test_Y = z.T
    print("Testing accuracy: {}".format(
        round(nn.evaluate({
            "features": test_features,
            "labels": test_Y
        }), 4)))
    print("Testing time: {0} seconds".format(round(time() - t, 2)))
Ejemplo n.º 2
0
def xor_net():
    a = np.array([[0, 0, 0], [0, 1, 1], [1, 0, 1], [1, 1, 0]])
    features = a[:, 0:2].T
    labels = a[:, 2].reshape((1, 4))
    z = np.zeros((4, 2))
    z[np.arange(4), labels] = 1
    labels = z.T
    nn = NN([2, 4, 3, 2])
    nn.set_hyperparameters(batch_size=4, learning_rate=0.75)
    t = time()
    nn.initialize_parameters()
    print("Start Training...")
    nn.minimize({"features": features, "labels": labels}, 10000)
    print("Finish Training.")
    print("Training time: {0} seconds".format(round(time() - t, 2)))
    print("Start Testing...")
    t = time()
    print("Testing accuracy: {}".format(
        round(nn.evaluate({
            "features": features,
            "labels": labels
        }), 4)))
    print("Testing time: {0} seconds".format(round(time() - t, 2)))