将 PyNeural 模型保存到磁盘
Saving a PyNeural model to disk
我使用 PyNeural 训练了一个模型。
如何保存这个训练好的模型?
我尝试了 pickle
和 sklearn.externals.joblib
但没有用。
至于后面说的TypeError: can't pickle NeuralNet objects
这里是 PyNeural 的作者。最简单的方法是调用
NeuralNet.get_params()
NeuralNet 对象上的方法。这将 return 一个由模型参数组成的元组。您应该能够 pickle 和 un-pickle 这个元组。要将参数加载到新模型中,实例化一个具有相同大小和层数的新 NeuralNet 对象,然后调用
NeuralNet.set_params(params)
其中 'params' 是您保存的元组。
希望您喜欢使用 PyNeural!
我使用 PyNeural 训练了一个模型。
如何保存这个训练好的模型?
我尝试了 pickle
和 sklearn.externals.joblib
但没有用。
至于后面说的TypeError: can't pickle NeuralNet objects
这里是 PyNeural 的作者。最简单的方法是调用
NeuralNet.get_params()
NeuralNet 对象上的方法。这将 return 一个由模型参数组成的元组。您应该能够 pickle 和 un-pickle 这个元组。要将参数加载到新模型中,实例化一个具有相同大小和层数的新 NeuralNet 对象,然后调用
NeuralNet.set_params(params)
其中 'params' 是您保存的元组。
希望您喜欢使用 PyNeural!