我正在使用Keras将我的Bi-Lstm模型保存在Google Colab上,代码如下
with open('BiLstm_model.pkl', 'wb') as writer:
pickle.dump(lstm_model, writer)
现在,当我在本地PC中使用代码加载此模型时,它显示了图像中显示的错误。
lstm_model = pickle.load(open('BiLstm_model.pkl', 'rb'))
Error Message
每当我在Google colab中加载模型时,它都在工作,但是当我在本地PC中加载它时,它会给我错误。我已经尝试使用不同的格式和库来保存模型并加载它。此外,我已经安装了相同版本的包指定和使用的谷歌可乐。
1条答案
按热度按时间dy1byipe1#
解决方案是尝试使用hdf5保存和加载模型。下面是代码片段
并使用