This repo contains the notebook, code, and requirements for training a chest X-ray classification model.
- Create a Python environment
- Install dependencies:
pip install -r requirements.txt - Run the notebook
train_eval.ipynb
- Learning rate: 0.001
- Batch size: 32
- Epochs: 10