This neural network uses your GPU to train on the MNIST dataset and learn to recognize images of hand-written digits.
If you don't have a GPU or don't have CuPy installed on your system, the program will use NumPy instead. The training will run much slower, but it will achieve the same results.
Email [email protected] with questions.
Video Demo in my AI Finance series
Enjoy!
- Python
- NumPy
- CuPy (if you want to use your GPU)
- Requests (unless you already have the csv files: train & test)
- Matplotlib
git clone https://github.com/chivington/Neural-Network-Performance-Analyzer.gitFrom the program folder, just run:
python mnist-nn.py...and follow the prompts.
For more details, see the Video Demo
Feel free to ask me questions on GitHub or at [email protected]
Not currently accepting outside contributors, but feel free to use as you wish.
There is no license associated with this content.