neural_network.py

from numpy import exp, array, random, dot


class NeuralNetwork:
    def __init__(self):
        random.seed(1)
        self.weights = 2 * random.random((3, 1)) - 1

    def __sigmoid(self, x):
        return 1 / (1 + exp(-x))

    def train(self, inputs, outputs, num):
        for iteration in range(num):
            output = self.think(inputs)
            error = outputs - output
            adjustment = dot(inputs.T, error * output * (1 - output))
            self.weights += adjustment

    def think(self, inputs):
        result = self.__sigmoid(dot(inputs, self.weights))
        return result


network = NeuralNetwork()

inputs = array([[1, 1, 1], [1, 0, 1], [0, 1, 1]])
outputs = array([[1, 1, 0]]).T

network.train(inputs, outputs, 10000)

result = network.think(array([1, 0, 0]))

if result >= 0.5:
    print(1)
else:
    print(0)