如何保存在谷歌datalab笔记本中训练的张量流模型进行离线预测?

时间:2017-06-07 10:08:55

标签: tensorflow google-cloud-platform google-cloud-datalab

我正在使用Google Cloud Datalab笔记本来训练我的张量流模型。我想保存训练有素的离线预测模型。但是,我对如何保存模型毫无头绪。我应该使用任何张量流模型保存方法还是有任何datalab / google云存储特定方法这样做?在这方面的任何帮助都非常感谢。

1 个答案:

答案 0 :(得分:2)

您可以使用任何张量流模型保存方法,但我建议您将其保存到Google云端存储分区而不是本地磁盘。大多数tensorflow方法使用gs://前缀

接受Google云端存储路径代替文件名

我建议使用SavedModelBuilder,因为它目前是最便携的。这里有一个例子:dynamodb-geo