Keras：ImageDataGeneratorまたはpredict_generatorからTrueラベル（y_test）を取得

Question

ImageDataGenerator().flow_from_directory(...)を使用して、ディレクトリからデータのバッチを生成しています。

モデルが正常に構築されたら、真と予測クラスラベルの2列の配列を取得したいと思います。 model.predict_generator(validation_generator, steps=NUM_STEPS)を使用すると、予測クラスの数が少ない配列を得ることができます。 predict_generatorに対応するTrueクラスラベルを出力させることは可能ですか？

追加する：validation_generator.classesはTrueラベルを出力しますが、ディレクトリから取得する順番で、増強によるバッチ処理またはサンプル展開は考慮しません。

Answer 1

あなたがして予測ラベルを取得できます（予測） y_pred = numpy.rint、あなたはで真のラベルを取得することができます： y_true = validation_generator.classesあなたはこの前に検証ジェネレータでシャッフル= Falseに設定する必要があり 。

最後に、混同行列を印刷することができます。 印刷confusion_matrix（y_true、y_pred）