How to build a imbalanced dataset based on your code?

First, thanks so much for your work. In your code, we have to specify the argument "labels_per_class_test", but I have a imbalanced dataset, which means that the number of labels for each class is different, so what should I do?