将 PyNeural 模型保存到磁盘

Saving a PyNeural model to disk

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

如何保存这个训练好的模型? 我尝试了 picklesklearn.externals.joblib 但没有用。 至于后面说的TypeError: can't pickle NeuralNet objects

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

NeuralNet.get_params()

NeuralNet 对象上的方法。这将 return 一个由模型参数组成的元组。您应该能够 pickle 和 un-pickle 这个元组。要将参数加载到新模型中,实例化一个具有相同大小和层数的新 NeuralNet 对象,然后调用

NeuralNet.set_params(params)

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

希望您喜欢使用 PyNeural!