CNN Model on Chest X-ray Dataset

Building a CNN model on chest x-ray dataset. We use a smaller sample of the dataset

Add more data to get a better result

Dataset on kaggle

Contents

Data Augmentation

The data is huge, so we don't load the data directly

There are two classes and the data is imbalanced

Imbalanced Data Handling

Class weights

Modeling

We use the following architecture for our model

Model Improving

Early stopping
ReduceLROnPlateau

Evaluation

Accuracy and loss plotting