changements tests

This commit is contained in:
eynard 2021-12-17 08:36:45 +01:00
parent 619f4762ef
commit bfd06b4f29

View File

@ -4,37 +4,42 @@ from sobek.network import network
random.seed() random.seed()
myNetwork = network(1, 10) myNetwork = network(10, 10)
learningRate = 1 learningRate = 1
for j in range(100000): for j in range(10000):
inputs = [] inputs = []
inputs2 = []
desiredOutputs = [] desiredOutputs = []
if (j%50 == 0): if (j%50 == 0):
print(j) print(j)
for i in range(1000): for i in range(1000):
inputs.append([random.randrange(10)]) inputs.append([(random.randrange(10)/10)])
inputs = np.array(inputs, dtype=object) inputs = np.array(inputs, dtype=object)
for i in range(1000): for i in range(1000):
desiredOutputs.append([0]*10) desiredOutputs.append([0]*10)
desiredOutputs[i][9 - inputs[i][0]] = 1.0 desiredOutputs[i][9 - int(inputs[i][0]*10)] = 1.0
desiredOutputs = np.array(desiredOutputs, dtype=object) desiredOutputs = np.array(desiredOutputs, dtype=object)
for i in range(1000):
inputs2.append([0]*10)
inputs2[i][int(inputs[i][0]*10)] = 1.0
inputs2 = np.array(inputs2, dtype=object)
if (j%10000 == 0): if (j%10000 == 0):
learningRate*= 0.1 learningRate*= 0.1
myNetwork.train(inputs, desiredOutputs, learningRate)
print(myNetwork.process(np.array([0.0], dtype=object))) myNetwork.train(inputs2, desiredOutputs, learningRate)
print(myNetwork.process(np.array([1.0], dtype=object)))
print(myNetwork.process(np.array([2.0], dtype=object))) test = []
print(myNetwork.process(np.array([3.0], dtype=object))) test.append([0]*10)
print(myNetwork.process(np.array([4.0], dtype=object))) test.append([0]*10)
print(myNetwork.process(np.array([5.0], dtype=object))) test[0][1] = 1.0
print(myNetwork.process(np.array([6.0], dtype=object))) test[1][8] = 1.0
print(myNetwork.process(np.array([7.0], dtype=object))) test = np.array(test, dtype=object)
print(myNetwork.process(np.array([8.0], dtype=object))) print(myNetwork.process(test[0]))
print(myNetwork.process(np.array([9.0], dtype=object))) print(myNetwork.process(test[1]))