Introductory_Deep_Learning_Classifier

Starter kit for Deep Learning Image Classification

Take a look at the notebooks for more.

Background

Data: MedNIST Dataset

Model Layout: Input -> Conv2D -> ReLU -> MaxPool2D -> Conv2D -> ReLU -> MaxPool2D -> FC1 -> FC2 -> FC3 -> Output

Loss Calculator: Cross-Entropy Loss

Optimizer: Adam

Requirements

1. NumPy

2. scikit-learn

3. Matplotlib

4. PyTorch