Generates stratified test-train splits for multi-label data.