-
-
Notifications
You must be signed in to change notification settings - Fork 243
Add Mira sex prediction model #173
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Pull Request Overview
This PR adds a MIRA sex prediction model for predicting patient sex from chest X-ray images. The model implements a ResNet34 architecture trained on the CheXpert dataset and provides binary classification between "Male" and "Female" categories.
- Implements a new SexModel class within the MIRA baseline models module
- Adds comprehensive test coverage for the new model functionality
- Integrates the MIRA module into the baseline models package structure
Reviewed Changes
Copilot reviewed 5 out of 6 changed files in this pull request and generated 3 comments.
Show a summary per file
| File | Description |
|---|---|
| torchxrayvision/baseline_models/mira/init.py | New SexModel implementation with ResNet34 architecture and weight loading |
| torchxrayvision/baseline_models/init.py | Imports the new mira module |
| tests/test_mira_sex_comprehensive.py | Comprehensive test suite for the SexModel functionality |
| tests/test_covid_dataloader.py | Adds skip markers to existing tests |
| tests/test_baseline_models.py | Adds tests for the new MIRA sex model |
Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.
From: https://www.thelancet.com/journals/ebiom/article/PIIS2352-3964(23)00032-4/fulltext