PT21-22-Reseau-Neurones/sobek/network.py

105 lines
4.5 KiB
Python
Raw Normal View History

2021-12-02 17:34:04 +01:00
import numpy as np
2021-12-16 17:06:51 +01:00
import math
2021-12-02 17:34:04 +01:00
class network:
def __init__(self, inputLayerSize, *layerSizes):
2021-12-03 15:10:27 +01:00
if type(inputLayerSize) != int:
raise TypeError("The input layer size must be an int!")
2021-12-09 18:50:57 +01:00
self.__weights = []
self.__inputLayerSize = inputLayerSize
oldLayerSize = inputLayerSize
2021-12-02 17:34:04 +01:00
for layerSize in layerSizes:
2021-12-16 17:06:51 +01:00
self.__weights.append( np.random.random((layerSize, oldLayerSize)) )
2021-12-09 18:50:57 +01:00
oldLayerSize = layerSize
self.__biases = [[0]*layerSize for layerSize in layerSizes]
self.__weights = np.array(self.__weights, dtype=object)
self.__biases = np.array(self.__biases, dtype=object)
2021-12-02 17:34:04 +01:00
2021-12-14 10:44:48 +01:00
def __reLu(value, derivative=False):
if (derivative):
return 0 if (value == 0) else 1
2021-12-02 17:34:04 +01:00
return max(0, value)
2021-12-14 10:44:48 +01:00
def __sigmoid(value, derivative=False):
if (derivative):
return network.__sigmoid(value) * (1 - network.__sigmoid(value))
2021-12-16 17:06:51 +01:00
return 1/(1+math.exp(-value))
2021-12-14 10:44:48 +01:00
def process(self, _input, __storeValues=False):
if type(_input) != np.ndarray:
2021-12-03 15:10:27 +01:00
raise TypeError("The input must be a vector!")
if _input.size != self.__inputLayerSize:
2021-12-03 15:10:27 +01:00
raise ValueError("The input vector has the wrong size!")
#if _input.dtype != np.float64:
# raise TypeError("The input vector must contain floats!")
if (__storeValues):
2021-12-16 17:06:51 +01:00
self.activations = []
self.outputs = []
2021-12-03 15:10:27 +01:00
2021-12-09 18:50:57 +01:00
for layerWeights, bias in zip(self.__weights, self.__biases):
2021-12-16 17:06:51 +01:00
_input = np.matmul(layerWeights, _input)
_input = np.add(_input, bias)
if (__storeValues):
2021-12-16 17:06:51 +01:00
self.activations.append(_input.copy())
2021-12-03 15:10:27 +01:00
#reLu application
2021-12-16 17:06:51 +01:00
for neuron in range(len(_input)):
_input[neuron] = network.__sigmoid(_input[neuron])
#On peut comparer la performance si on recalcul plus tard
if (__storeValues):
2021-12-16 17:06:51 +01:00
self.outputs.append(_input.copy())
self.activations = np.array(self.activations, dtype=object)
self.outputs = np.array(self.outputs, dtype=object)
2021-12-03 15:10:27 +01:00
return _input
2021-12-09 18:50:57 +01:00
2021-12-16 17:06:51 +01:00
def train(self, inputs, desiredOutputs, learningRate):
2021-12-16 17:06:51 +01:00
errorSums = [[[0]*(len(neuron)) for neuron in layer] for layer in self.__weights]
self.__errors = [[0]*(len(layer)) for layer in self.__weights]
for _input, desiredOutput in zip(inputs, desiredOutputs):
self.__output = self.process(_input, True)
self.__desiredOutput = desiredOutput
2021-12-16 17:06:51 +01:00
for layerNumber in range(len(errorSums)-1, -1, -1):
for neuronNumber in range(len(errorSums[layerNumber])):
for weightNumber in range(len(errorSums[layerNumber][neuronNumber])):
#print("layer : " + str(layerNumber) + " neuron : " + str(neuronNumber) + " weight : " + str(weightNumber))
errorSums[layerNumber][neuronNumber][weightNumber] += self.__partialDerivative(layerNumber, neuronNumber, weightNumber)
total = 0
for i in range(len(errorSums)):
for j in range(len(errorSums[i])):
for k in range(len(errorSums[i][j])):
errorSums[i][j][k] = errorSums[i][j][k] / len(inputs)
total += errorSums[i][j][k]
self.__weights[i][j][k] -= learningRate * errorSums[i][j][k]
print("Error : " + str(total))
2021-12-14 10:44:48 +01:00
def __Error(self, layer, neuron):
2021-12-16 17:06:51 +01:00
if (self.__errors[layer][neuron] == 0 ):
self.__errors[layer][neuron] = self.__ErrorFinalLayer(neuron) if (layer == len(self.__weights)-1) else self.__ErrorHiddenLayer(layer, neuron)
return self.__errors[layer][neuron]
2021-12-14 10:44:48 +01:00
def __ErrorFinalLayer(self, neuron):
2021-12-16 17:06:51 +01:00
return network.__sigmoid(self.activations[len(self.activations)-1][neuron], True) * (self.__output[neuron] - self.__desiredOutput[neuron])
def __ErrorHiddenLayer(self, layer, neuron):
upperLayerLinksSum = 0
for upperLayerNeuron in range(len(self.__weights[layer+1]-1)):
#A comparer avec un acces direct au erreurs precalcules
upperLayerLinksSum += self.__weights[layer+1][upperLayerNeuron][neuron] * self.__Error(layer+1, neuron)
2021-12-16 17:06:51 +01:00
return network.__sigmoid(self.activations[layer][neuron], True) * upperLayerLinksSum
2021-12-16 17:06:51 +01:00
def __partialDerivative(self, layer, neuron, weight):
return self.__Error(layer, neuron) * self.outputs[layer-1][weight]