我有一个关于机器学习的非常基础的问题。我一直在学习各种机器学习教程和在线课程,了解如何使用各种技术,但没有看到关于如何持久化应用的部分。
例如,我训练了一个网络来识别花园侏儒的图像,但在运行训练集并用测试数据验证后,我该如何保存这个网络,以便将来可以输入一张图片,让它告诉我这张图片是否是花园侏儒?每个教程似乎都只让你运行训练/验证集,却没有提到如何以有意义的方式托管网络以供将来使用。
谢谢!
回答:
一旦模型/网络通过测试/开发集训练和验证后,最佳实践是“冻结”其状态,并用于预测。
为了将网络或模型集成到应用程序中,应该保持模型的状态并存储它,通常是通过序列化、标记并保存到有状态的存储(数据库或本地磁盘)中来存储模型的。