我使用PyNeural训练了一个模型。
如何保存此训练模型?
我尝试了pickle
和sklearn.externals.joblib
,但没有成功。
如后,说TypeError: can't pickle NeuralNet objects
答案 0 :(得分:1)
PyNeural的作者在这里。最简单的方法是调用
NeuralNet.get_params()
神经网络对象上的方法。这将返回由模型参数组成的元组。你应该能够腌制和取消这个元组。要将参数加载到新模型中,请实例化具有相同大小和层数的新NeuralNet对象,然后调用
NeuralNet.set_params(params)
其中' params'是你保存的元组。
希望你喜欢使用PyNeural!