Exclude pediatric patients (younger than 18) from clinical prediction tasks#1150
Exclude pediatric patients (younger than 18) from clinical prediction tasks#1150DarylOkeke wants to merge 3 commits into
Conversation
jhnwu3
left a comment
There was a problem hiding this comment.
Hey nice first PR!
Is it possible to add a Boolean arg for this filtering? I think it'd be nice to be able to quickly ablation study this difference in patient demographics later.
|
Thanks! I added exclude_minors to all the affected classes and legacy functions in the 3 task files. Defaults to true so the existing behavior stays the same Another thing though, I saw that LengthOfStayPredictioneICU doesn't have an age check at all. The original TODOs didn't cover it but the other 3 LOS classes do. Want me to add one for consistency, or no? |
|
Yeah that would be nice |
|
Finished! |
|
Will check back after unit tests finish this weekend |
jhnwu3
left a comment
There was a problem hiding this comment.
One last thing before we meet, can you verify it works locally on your machine with our demo dataset?
|
closed bc duplicate (it seems?). Reopen if you disagree |
Multiple clinical prediction tasks had TODO comments saying that patients under 18 should be excluded, but the filtering was never implemented.
This adds age filtering to the mortality prediction, length-of-stay, and drug recommendation tasks across MIMIC-III, MIMIC-IV, eICU, and OMOP datasets. Adult prediction benchmarks shouldn't mix in pediatric cohorts b/c the clinical patterns are too different.
Some notes: