---
datasets:
- uraninjo/augmented-alzheimer-mri-dataset
- matthewchung74/alzheimer-mri-disease-classification-dataset
language: en
license: apache-2.0
metrics:
- name: accuracy
  value: 0.8833869769744986
- name: f1
  value: 0.8819737103520164
model_name: Baseline CNN
tags:
- medical
- mri
- alzheimer
- classification
- pytorch
---
# Baseline CNN

## Performance
- **Accuracy**: 0.8834
- **F1 Score**: 0.8820

## Resources
- **Colab Notebook**: [Open in Colab](https://colab.research.google.com/drive/1xwKuym6RYXj-nF2HeJueSRHIyadtuiSG?usp=sharing)
- **Datasets Used**:
    - [Kaggle: Augmented Alzheimer MRI Dataset](https://www.kaggle.com/datasets/uraninjo/augmented-alzheimer-mri-dataset)
    - [Hugging Face: Alzheimer MRI Disease Classification Dataset](https://huggingface.co/datasets/matthewchung74/alzheimer-mri-disease-classification-dataset)

## Classification Report
```
                  precision    recall  f1-score   support

     NonDemented       0.82      0.93      0.87      1280
VeryMildDemented       0.89      0.75      0.81      1120
    MildDemented       0.89      0.91      0.90       986
ModerateDemented       0.99      1.00      0.99       653

        accuracy                           0.88      4039
       macro avg       0.90      0.89      0.89      4039
    weighted avg       0.89      0.88      0.88      4039

```