保存和加载后,keras模型的预测是难得的

时间:2020-01-15 22:40:11

标签: python tensorflow keras google-colaboratory

我用google colab训练了神经网络。
我使用joblib.dump()

保存了神经网络

然后我使用joblib.load()

将模型加载到PC上

我在colab和PC上使用相同的模型对完全相同的样本进行了预测。在colab上,其输出为[[0.51]]。在我的电脑上,其输出为[[nan]]

该模型摘要报告该模型的体系结构是相同的。 我检查了PC上加载的模型的重量以及colab上的模型的重量,这些重量完全相同。

关于我能做什么的任何想法?谢谢。

快速更新:即使我将所有输入都更改为零,预测也仍然是

1 个答案:

答案 0 :(得分:0)

据我所知,keras具有自己的功能来保存诸如model.save('file.h5')之类的模型,而joblib库则用于保存sklearn模型。