Would be great to have the train test split function in LegateBoost for convenience before we can find a host for a formal HPO procedures.