Group labels by the same color

Feature description

Hi, my sequence labeling is fine-grained, I want to group labels by the same color, where can I find the code to change?

Thanks for the great tool!