如何获取张量流的预先估计器的图定义

时间:2018-06-22 18:22:48

标签: python tensorflow

作为tensorflow的初学者,我很好奇看到tensorflow的预先估计器(例如tf.estimator.LinearClassifier)定义的图形。

具有以下代码段:

import numpy as np
import tensorflow as tf

estimator = tf.estimator.LinearClassifier(    
    feature_columns=[tf.feature_column.numeric_column("x")],
    model_dir="stackoverflow_model"
)

input_fn = tf.estimator.inputs.numpy_input_fn(x={"x": np.arange(5)}, y=np.array([1] * 5), shuffle=True)

estimator.train(input_fn)

我可以在张量板上看到LinearClassifier定义的图形:

enter image description here

但是,如果我什么都没得到:

print(tf.get_default_graph().as_graph_def())

我只会得到:

versions {
  producer: 26
}

所以我的问题是:LinearClassifier定义的图形在哪里?如何正确检查using as_graph_def()

谢谢!

0 个答案:

没有答案