keras 通过pickle加载深度学习模型时出错

dbf7pr2w 于 2022-11-13 发布在其他

关注(0)|答案(1)|浏览(178)

我使用Keras制作了一个深度学习模型，并将其存储在一个名为model.pkl的文件夹中，为了加载部署模型，我使用了代码，即

import pickle
model = pickle.load(open('/home/samar/Desktop/ckd/model.pkl', 'rb'))
prediction = model.predict(data)

但它返回我

IsADirectoryError: [Errno 21] Is a directory: '/home/samar/Desktop/ckd/model.pkl'

虽然文件夹存在，但仍会出现错误。
我通过model.save('model.pkl')保存了模型

1条答案

使用model.save()方法实际上并不会对它进行pickle操作，您可能希望使用keras中内置的模型加载器来加载模型，如下所示：

model = keras.models.load_model('/home/samar/Desktop/ckd/model.pkl')
prediction = model.predict(data)