Question

我正在尝试将我的模型导出到 Google云端存储。我使用tf.contrib.learn来构建我的模型并遵循虹膜分类示例。

在我的培训和评估完成后，我想将模型存储在云端，以便我可以进行预测，但我不知道如何导出模型。

classifier = tf.contrib.learn.DNNClassifier(feature_columns=feature_columns,
                                            hidden_units=[100],
                                            n_classes=50,
                                            model_dir="Model_Logs")

Answer 1

云上培训的最佳示例可能是census (canned estimator)或census (custom estimator)。它们使用相同的Estimator API，因此该部分应该是熟悉的。此外，他们使用Estimator课程来帮助自动执行培训。 train_and_evaluate方法会在该类上调用learn_runner.run方法，如果配置正确，将导出模型，这基本上归结为设置export_strategy和model_dir

如果您想在Experiment和learn_runner框架之外执行操作，只需拨打Estimator.export_savedmodel

即可

将Tensorflow模型导出到Google云端存储

1 个答案: