keras/tensorflow 模型保存后重新加载准确率为0 model.save and load giving different result

我在用别人的代码跑程序的时候遇到了这个问题:
keras 模型保存后重新加载准确率为0
GitHub上有个issue:model.save and load giving different result · Issue #4875 · keras-team/keras
这个issue有150个回复了,大部分人都是遇到了这个问题。
我用tensorflow重写了一遍,没有解决问题。
然后我怀疑不是库的问题,应该是其他部分代码的问题。
最终我发现问题是在加载数据时使用了set去重,但是set是无序的,所以每次加载数据的标签号都不一样。把set转换成list然后排序一下就解决了问题。
解决这个问题花了我整整2天。

猜你喜欢

转载自blog.csdn.net/zhangpeterx/article/details/89353354