matthewchung74/baseline-cnn-augmented-alzheimer

Baseline CNN Augmented for Alzheimer's MRI Classification

This model predicts the severity of Alzheimer's disease from MRI scans. It was trained on the Alzheimer MRI Disease Classification Dataset.

Performance (Test Set)

• Accuracy: 0.5539
• Weighted F1-Score: 0.5177

Classification Report

                    precision    recall  f1-score   support

     Mild Demented       0.00      0.00      0.00       172
 Moderate Demented       0.00      0.00      0.00        15
      Non Demented       0.79      0.53      0.63       634
Very Mild Demented       0.44      0.81      0.57       459

          accuracy                           0.55      1280
         macro avg       0.31      0.34      0.30      1280
      weighted avg       0.55      0.55      0.52      1280

How to use

This is a PyTorch model. Load using torch.load.