Machine learning stratification for oncology patient survival