python - 训练卷积神经网络时如何平衡keras上的小批量？

Question

我正在训练图像上的卷积神经网络（大小为 299、299、3）。图像可以有标签：0、1 或 2（多类分类），这 3 个类非常不平衡。为了改进训练阶段，我想确保每个 minibatch 包含所有三个类。我怎样才能做到这一点？

我正在使用 flow_from_dataframe 将图像传递给神经网络。

train_generator=datagen.flow_from_dataframe(
dataframe=train, 
directory=None, 
x_col="directory",
y_col="label",
batch_size=64, 
seed=42, 
shuffle=False, 
class_mode='categorical',
target_size=(299,299))