hyeonsangjeon / neural-network

Machine Learning library for educational purpose.

Neural Network From Scratch

This code is part of my video series on YouTube: Neural Network from Scratch | Mathematics & Python Code.

Try it!

python3 xor.py

Example

import numpy as np

from dense import Dense
from activations import Tanh
from losses import mse, mse_prime
from network import train

X = np.reshape([[0, 0], [0, 1], [1, 0], [1, 1]], (4, 2, 1))
Y = np.reshape([[0], [1], [1], [0]], (4, 1, 1))

network = [
    Dense(2, 3),
    Tanh(),
    Dense(3, 1),
    Tanh()
]

train(network, mse, mse_prime, X, Y, epochs=10000, learning_rate=0.1)