Skip to content

Latest commit

 

History

History
18 lines (14 loc) · 358 Bytes

File metadata and controls

18 lines (14 loc) · 358 Bytes

Neural Network in pure Numpy

Implemented a fully connected neural network in pure NumPy and compared the performance with Pytorch on the Iris and MNIST dataset.

Run

# use nn implemented here
python train.py

# use nn in pytorch
python pytorch_train.py

Accuracy

nn-numpy nn-pytorch
Iris 0.978 0.956
MNIST 0.943 0.974