balanced data set because you've increased this K parameter so they show how there is a limiting effect to how large you increase the K because naturally as you increase the K passed a certain threshold you're making your knowledge distillation data step four your student network to be really imbalanced and deep learning machine learning these kinds of decision boundary models do not respond well to class and balance although they don't show you like things like random over sampling a lot of the techniques commonly used to overcome class