imbalanced training dataset for uGB