将PyNeural模型保存到磁盘

时间:2015-05-26 12:49:49

标签: python joblib

我使用PyNeural训练了一个模型。

如何保存此训练模型? 我尝试了picklesklearn.externals.joblib,但没有成功。 如后,说TypeError: can't pickle NeuralNet objects

1 个答案:

答案 0 :(得分:1)

PyNeural的作者在这里。最简单的方法是调用

NeuralNet.get_params()
神经网络对象上的

方法。这将返回由模型参数组成的元组。你应该能够腌制和取消这个元组。要将参数加载到新模型中,请实例化具有相同大小和层数的新NeuralNet对象,然后调用

NeuralNet.set_params(params)

其中' params'是你保存的元组。

希望你喜欢使用PyNeural!