CNS-lab

To run the code. Clone this repository.

Open in IDE that supports Jupyter Notebook or Google Colab

Make sure the digits_X.npy and digits_X.npy are in the same working directory as the nns.ipynb

Required packages

numpy, matplotlib, torch, torchvision

(OPTIONAL) Create a Conda environment

in Conda prompt, specify python version

conda create -n EnvName python=3.x

conda activate EnvName

pip install numpy matplotlib torch torchvision

Select the Conda environment as the Kernel for running the notebook in your IDE