Imported from https://github.com/tonyduan/hte-prediction-rcts