Skip to content

Machine learning project for classifying chest X-rays as healthy or pneumonia-affected, featuring multiple algorithms and imbalance handling techniques.

License

Notifications You must be signed in to change notification settings

titouanlegourrierec/PneumoniaXRayClassification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

LinkedIn Mail

Pneumonia XRay Classification

LE GOURRIEREC Titouan
Report a bug · Request Feature

Table of Contents
  1. About The Project
  2. Getting Started
  3. Usage
  4. License
  5. Contact

About The Project

This project focuses on classifying chest X-rays as either healthy or pneumonia-affected. Several machine learning algorithms were tested, including K-Nearest Neighbors (KNN), Logistic Regression, Decision Trees, Support Vector Classifier (SVC), Random Forest, Gradient Boosting Classifier, and a Voting Classifier.

To address the issue of imbalanced classes, various techniques were employed, such as Random Over Sampling, SMOTE (Synthetic Minority Over-sampling Technique), Random Under Sampling, and Weight Modification.

For each algorithm, Grid Search was used to fine-tune the hyperparameters. The final model was evaluated using cross-validation, and learning curves were analyzed to assess whether further evaluation was needed.

The final model chosen is a soft voting classifier, which combines the outputs of the different models, configured as follows:

Below are the evaluation metrics for this model:

You can find the project report here: report.pdf

(back to top)

Built With

  • Python
  • OpenCV
  • scikit-learn

(back to top)

Getting Started

Prerequisites

Before running this project, make sure you have installed the necessary dependencies. You can do this by installing the packages listed in the requirements.txt file:

pip install -r requirements.txt

Ensure you have Python installed and that you're using a virtual environment if needed.

-> For the data, please follow the instructions in the Dataset_Link.pdf file.

(back to top)

Usage

To use this project, just change the variable path in the Images Import part of the file project.ipynb

(back to top)

License

Distributed under the MIT License. See LICENSE for more information.

(back to top)

Contact

LE GOURRIEREC Titouan - titouanlegourrierec@icloud.com

Project Link: https://github.com/titouanlegourrierec/PneumoniaXRayClassification

(back to top)

About

Machine learning project for classifying chest X-rays as healthy or pneumonia-affected, featuring multiple algorithms and imbalance handling techniques.

Topics

Resources

License

Stars

Watchers

Forks