Keras flow_from_directory class index

后端 未结 3 1954
别那么骄傲
别那么骄傲 2021-02-08 08:43

I used to make it manually, but i am using now flow_from_directory to train my network with my own data. I just have one question. When i make model.predict(), how can i know th

3条回答
  •  孤城傲影
    2021-02-08 09:30

    You can have the the index of each class generated by the generator with class_indices property.

    print(validation_generator.class_indices)
    

    Simple...

提交回复
热议问题