Esempio n. 1
0
import os
os.chdir('..')

xys = []
nums = [0, 1, 3, 2, 6, 14, 10, 11, 9, 8, 12, 4, 5, 7, 15, 13]
num2 = [0, 1, 5, 4, 12, 8, 9, 13, 15, 11, 10, 14, 6, 7, 3, 2]
num3 = [0, 4, 12, 8, 9, 13, 5, 1, 3, 7, 15, 11, 10, 14, 6, 2]
num4 = [0, 8, 10, 2, 3, 7, 6, 4, 5, 1, 9, 13, 12, 14, 15, 11]
for i in range(16):
    x = np.binary_repr(i, 4)
    x = [int(j) for j in x]
    y = np.binary_repr(num4[i], 4)
    y = [int(j) for j in y]
    xys.append(
        (np.array(x).astype(np.float32), np.array(y).astype(np.float32)))

nn = NeuralNet(Layer(4, activation_type=TANH),
               Layer(64, activation_type=SIGMOID), Output(4))

nn.predict_values(xys)
fun = lambda: nn.train(xys,
                       batch_size=len(xys),
                       num_epochs=100,
                       print_every=100,
                       predict_after_every_batch=False)

fun()

print("**************FINAL PREDICTION***************************")
nn.predict_values(xys)