PT21-22-Reseau-Neurones/testLearning.py
2021-12-16 17:06:51 +01:00

28 lines
672 B
Python

import numpy as np
import random
from sobek.network import network
random.seed()
myNetwork = network(1, 8, 8, 10)
for j in range(3000):
inputs = []
desiredOutputs = []
if (j%50 == 0):
print(j)
for i in range(200):
inputs.append([random.randrange(10)])
inputs = np.array(inputs, dtype=object)
for i in range(200):
desiredOutputs.append([0]*10)
desiredOutputs[i][9 - inputs[i][0]] = 1
desiredOutputs = np.array(desiredOutputs, dtype=object)
myNetwork.train(inputs, desiredOutputs, 0.01)
print(myNetwork.process(np.array([8.0], dtype=object)))
print(myNetwork.process(np.array([1.0], dtype=object)))