2

我正在尝试将我的模型导出到Google Cloud Storage。我曾经tf.contrib.learn构建我的模型并遵循虹膜分类示例。

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

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

1 回答 1

0

云上训练的最佳示例可能是人口普查（罐装估算器）或人口普查（自定义估算器）。他们使用相同的 Estimator API，所以这部分应该很熟悉。此外，他们使用该Estimator课程来帮助自动执行培训。该train_and_evaluate方法由该类调用，learn_runner.run如果配置正确，它将导出模型，这基本上归结为设置export_strategy和model_dir

如果你想在Experimentandlearn_runner框架之外做事情，你可以调用Estimator.export_savedmodel

于 2017-06-26T16:50:31.533 回答