画像 (サイズ 299、299、3) で畳み込みニューラル ネットワークをトレーニングしています。画像には 0、1、または 2 (マルチクラス分類) のラベルを付けることができ、3 つのクラスは非常に不均衡です。トレーニング フェーズを改善するために、各ミニバッチに 3 つのクラスがすべて含まれていることを確認します。これどうやってするの？
画像をニューラル ネットワークに渡すために、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))