gpt4 book ai didi

python - 将 PyNeural 模型保存到磁盘

转载 作者:太空宇宙 更新时间:2023-11-03 17:41:38 25 4
gpt4 key购买 nike

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

如何保存这个经过训练的模型?我尝试了pickle和sklearn.externals.joblib但没有成功。正如后者所说的TypeError: can't pickle NeuralNet objects

最佳答案

PyNeural 的作者在这里。最简单的事情就是调用

NeuralNet.get_params()

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

NeuralNet.set_params(params)

其中“params”是您保存的元组。

希望您喜欢使用 PyNeural!

关于python - 将 PyNeural 模型保存到磁盘,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/30459088/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com