/CNN-Model-for-Pneumonia-Diagnosis

This repository contains a CNN model for pneumonia diagnosis using chest X-ray images. The model achieves 90.54% accuracy on the test dataset.

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

CNN Model for Pneumonia Diagnosis

Data Sample

This project focuses on developing a CNN model for pneumonia diagnosis using chest X-ray images. The model achieved an accuracy of 90.54% on the test dataset.

Dataset

  • Train dataset: 5216 samples
  • Validation dataset: 16 samples
  • Test dataset: 624 samples

The dataset used for this project can be found here.

Model Architecture

The model consists of convolutional layers, batch normalization, activation functions, max pooling, flattening, dense layers, dropout, and an output layer.

Training and Evaluation

  • Training loss: 0.2709
  • Training accuracy: 90.54%

Usage

To run the notebook and reproduce the results:

  1. Clone the repository
  2. Install dependencies
  3. Open the notebook file
  4. Run the notebook cells sequentially

The Kaggle notebook for this project can be found here.