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