保存和加载后keras模型预测为nan

Question

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

然后我使用joblib.load()将模型加载到我的电脑上

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

模型摘要报告模型的架构是相同的。 我检查了我在 PC 上加载的模型和 colab 上的模型的重量，重量完全一样。

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

快速更新：即使我将所有输入都更改为零，预测仍然是 nan。

Answer 1

据我所知keras有自己的函数保存模型如model.save('file.h5')，而joblib库是用来保存sklearn模型的。