Esempio n. 1
0
    def predict(self, data):
        nn = NeuralNetwork()
        l1 = Layer(56, 54)
        l2 = Layer(54, 25)

        nn.add(l1)
        nn.add(ActivationLayer(relu, relu_derivative))
        nn.add(l2)
        nn.add(ActivationLayer(sigmoid, sigmoid_derivative))

        l1.weights = np.load('weights1.npy')
        l2.weights = np.load('weights2.npy')

        l1.bias = np.load('bias1.npy')
        l2.bias = np.load('bias2.npy')

        out = nn.predict(data)
        pred = np.argmax(out)

        return pred