【发布时间】:2017-05-03 04:37:40
【问题描述】:
我了解 predict_generator 输出概率。为了得到这个类,我就找到了最大概率的索引,这将是最可能的类。但是我发现这样做之后,我得到的输出与调用 predict_classes 不同。我不明白为什么。有人可以解释一下吗?
【问题讨论】:
标签: machine-learning neural-network deep-learning keras conv-neural-network
我了解 predict_generator 输出概率。为了得到这个类,我就找到了最大概率的索引,这将是最可能的类。但是我发现这样做之后,我得到的输出与调用 predict_classes 不同。我不明白为什么。有人可以解释一下吗?
【问题讨论】:
标签: machine-learning neural-network deep-learning keras conv-neural-network
Keras 中的生成器使用 glob 列出按字母顺序排序的文件夹,您可以使用
获取训练期间使用的类# save classes to JSON
class_json = json.dumps(train_generator.class_indices)
with open("class.json", "w") as class_file:
class_file.write(class_json)
在批次生成器(此处)中对样本进行混洗,以便在 fit_generator 或评估生成器请求批次时给出随机样本。
如果对图像执行此操作,另一种可能性是不使用 https://github.com/fchollet/keras/issues/3477 中提到的 ImageDataGenerator 中的 rescale=1./255
希望有所帮助!
【讨论】: