Skip to content

Latest commit

 

History

History
38 lines (18 loc) · 1.1 KB

README.md

File metadata and controls

38 lines (18 loc) · 1.1 KB

MLP-vs-CNN

Comparing the performance of MLP and CNN on USPS dataset and visualizing it via TensorBoard.

Language used: Python

Libraries used: torch, torchvision, sklearn

Dataset: USPS dataset

Dataset Visualization:

Observations:

Model Epochs Train Loss Test Accuracy Precision Recall
MLP 10 0.0373 0.9312 0.9265 0.9243
CNN 10 0.0060 0.9497 0.9470 0.9449

Confusion Matrix for MLP and CNN:

Precision and Recall: