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: