我用google colab训练了神经网络。
我使用joblib.dump()
然后我使用joblib.load()
我在colab和PC上使用相同的模型对完全相同的样本进行了预测。在colab上,其输出为[[0.51]]
。在我的电脑上,其输出为[[nan]]
。
该模型摘要报告该模型的体系结构是相同的。 我检查了PC上加载的模型的重量以及colab上的模型的重量,这些重量完全相同。
关于我能做什么的任何想法?谢谢。
快速更新:即使我将所有输入都更改为零,预测也仍然是 。
答案 0 :(得分:0)
据我所知,keras具有自己的功能来保存诸如model.save('file.h5')
之类的模型,而joblib
库则用于保存sklearn模型。