如何使用Python将使用Keras准备的图像分类器之类的对象保存到任何文件中？

Question

我有一个使用keras创建的对象。我想将其保存到文件（序列化）中，以便以后可以将其用于预测。 以下是我的对象： 分类器

我正在尝试使用泡菜，但这会给我以下错误

import pickle

file_pi = open('Dog_Cat_Identifier.obj', 'w')

pickle.dump(classifier, file_pi)

  

回溯（最近通话最近）：

     

文件“”，
中的第1行      pickle.dump（classifier，file_pi）

     

TypeError：无法腌制_thread.RLock对象

我能知道代码有什么问题吗？

Answer 1

您可以使用软件包提供的save方法保存keras模型

from keras.models import load_model

model.save('my_model.h5') 

model = load_model('my_model.h5')

您可以使用load_model()中的keras.models函数来加载分类器。