Kerasのデータジェネレータの使用中にラベル情報を取得する方法は？

Question

以下Keras Pythonコード使用している間：私はx_batchを持って、numpyのアレイをy_batch

for x_batch,y_batch in datagen.flow_from_directory(
    directory = os.path.join(dataset_root_path,dataset_train_path), 
    target_size = (520,520), 
    class_mode = 'binary', 
    batch_size = 1 
): 

をIしかし、 "バイナリ" class_modeを使用していますので、y_batchのnumpyのアレイは数0.0または1.0に符号化され、このようにして、私はそのサンプルの本当のラベル、例えば "cat"や "dog"についての情報を失った。 '1.0'と '0.0'の出力に従ってラベル情報を取得するにはどうすればよいですか？

Answer 1

私はdataGeneratorををインスタンス化し、fit_generatorで訓練することをお勧めします：

train_gen = datagen.flow_from_directory(...) 
model.fit_generator(train_gen, ...) 

あなたはその後、（他の属性の中で）train_gen.class_indicesを持つクラスインデックスにアクセスすることができます。