Convolutional neural networks

This repository contains code that demonstrates how to create and train a convolutional neural network (CNN) on the MNIST dataset.

Getting Started

• This project relies on python=3.12. It was not tested with different versions
• To view a list of extra required packages, see requirements.txt
• Clone the repository to your local machine
• Once you have, cd into this repo and create the virtual environment (assuming you have conda installed) via

conda env create -f environment.yml

• Enter the virtual environment with conda activate build-cnn-env
• Install MedMNIST with pip install --no-deps medmnist==3.0.1
• Install the packages in the repo root directory using pip install -e . (you only need the -e option if you intend to edit the source code in build_cnn/)

Example

See my website for examples on how to use this code.

Citation

If you use this project, please use the citation information provided by GitHub via the “Cite this repository” button or cite it as follows:

@software{build_cnn_2025,
  author = {Alberto Garcia},
  title = {Build CNN},
  year = {2025},
  publisher = {GitHub},
  url = {https://github.com/AJG91/build-cnn},
  license = {MIT}
}