小编xag*_*agg的帖子

如何在theano中保存/序列化训练有素的模型?

我按照加载和保存的文档保存了模型.

# saving trained model
f = file('models/simple_model.save', 'wb')
cPickle.dump(ca, f, protocol=cPickle.HIGHEST_PROTOCOL)
f.close()
Run Code Online (Sandbox Code Playgroud)

ca是一个训练有素的自动编码器.这是一个类的实例cA.从我构建的脚本和保存模型我可以调用ca.get_reconstructed_input(...),ca.get_hidden_values(...)没有任何问题.

在另一个脚本中,我尝试加载训练的模型.

# loading the trained model
model_file = file('models/simple_model.save', 'rb')
ca = cPickle.load(model_file)
model_file.close()
Run Code Online (Sandbox Code Playgroud)

我收到以下错误.

ca = cPickle.load(model_file)
Run Code Online (Sandbox Code Playgroud)

AttributeError:'module'对象没有属性'cA'

python serialization loading save theano

8
推荐指数
1
解决办法
5947
查看次数

标签 统计

loading ×1

python ×1

save ×1

serialization ×1

theano ×1