有没有一种方法可以不重新训练我的NN?

时间:2019-07-04 18:30:54

标签: machine-learning deep-learning pycharm

如果我仅需要进行一些预测,是否需要每次都重新训练我的NN?或者我可以,如果这很傻,请原谅我,“保存”培训并且只做测试? 目前,我正在使用Pycharm,但是我已经看到,使用Spyder等其他IDE,您可以执行选定的代码行,在这种情况下NN如何保持训练而无需重新训练? 抱歉,如果这些问题太幼稚。

1 个答案:

答案 0 :(得分:1)

否,您不需要每次都重新训练您的NN。只需将模型参数保存到文件中,然后加载以进行新的预测即可。

您是否正在使用Tensorflow或Keras之类的任何机器学习框架?在Keras中,这很容易实现,有两种方法,一种是您可以在训练期间使用Callbacks保存模型,第二种是可以使用your_model_name.save('file_name.h5')然后使用load_model('file_name.h5加载)做一些预测。使用your_model_name.prediction(x)

顺便说一下,there is a nice guide介绍如何正确保存完整的模型架构或模型权重。

编辑:对于这两种方法,您都可以使用load_model,非常简单!