DRW interview question

Coding challenge involving a prediction problem on a dataset