如何解决TypeError:不能为深度学习模型腌制weakref对象
当我运行pickle.dump(model,open('modelDL.pkl','wb'))时,我收到TypeError:无法腌制weakref对象。
我创建了一个深度学习模型,试图将其保存。 下面是模型。
model = Sequential()
model.add( Dense(30,activation='relu') )
model.add( Dropout(0.5) )
model.add( Dense(20,activation='relu') )
model.add( Dropout(0.5) )
model.add( Dense(1,activation='sigmoid') )
model.compile(optimizer='adam',loss='binary_crossentropy',metrics=['accuracy'])
解决方法
不能pickleweakref是因为深度学习模型太大,pickle只用于存储小模型
Use this : HDF5 用于存储大数据
from keras.models import load_model
model.save('my_model.h5') # creates a HDF5 file 'my_model.h5'
返回一个编译好的模型
与上一个相同
model1 = load_model('my_model.h5')
y_pred = model1.predict(x_test)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。