CNN预测结果

时间:2018-05-24 09:39:30

标签: python-3.x neural-network deep-learning prediction convolutional-neural-network

我正在处理以下链接 https://becominghuman.ai/building-an-image-classifier-using-deep-learning-in-python-totally-from-a-beginners-perspective-be8dbaf22dd8 我怎么知道“result [0] [0]”中的“one”是对“狗”而不是猫的预测?

1 个答案:

答案 0 :(得分:0)

本文中的代码使用ImageDataGenerator类和flow_from_directory()函数来加载和扩充图像。检查Keras documentation,您可以找到有关其中一个名为classes的参数的信息:

  

类子目录的可选列表(例如[' dogs',' cats'])。   默认值:无。如果未提供,则列表将为   从子目录名称/结构下自动推断出   目录,其中每个子目录将被视为不同的   class(以及将映射到标签的类的顺序   索引,将是字母数字)。包含映射的字典   从类名到类索引可以通过属性获得   class_indices。

根据这些信息,我怀疑是“猫”。是指一只0而一只狗#39;指的是1.确保你可以检查班级词典:

print(train_datagen.class_indices)