Stratifying patient cohorts