Not a blocker for this PR, but thinking ahead for ranking problems, is there a more robust way to distinguish classification v. regression? I imagine the labels for ranking would likely by integers as well. Or I guess I can cross that bridge when I get there.