Find centralized, trusted content and collaborate around the technologies you use most.
Teams
Q&A for work
Connect and share knowledge within a single location that is structured and easy to search.
我使用PyNeural训练了一个模型。
如何保存这个训练有素的模型?我试过了pickle,sklearn.externals.joblib但没有奏效。正如后来所说TypeError: can't pickle NeuralNet objects
pickle
sklearn.externals.joblib
TypeError: can't pickle NeuralNet objects
PyNeural 的作者在这里。最简单的方法是调用
NeuralNet.get_params()
您的 NeuralNet 对象上的方法。这将返回一个由模型参数组成的元组。您应该能够腌制和取消腌制这个元组。要将参数加载到新模型中,请实例化一个具有相同大小和层数的新 NeuralNet 对象,然后调用
NeuralNet.set_params(params)
其中 'params' 是您保存的元组。
希望您喜欢使用 PyNeural!