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

K.C*_*hen 5 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)
Run Code Online (Sandbox Code Playgroud)

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

在此处输入图片说明

但是,如果我什么都没有,则:

print(tf.get_default_graph().as_graph_def())
Run Code Online (Sandbox Code Playgroud)

我只会得到:

versions {
  producer: 26
}
Run Code Online (Sandbox Code Playgroud)

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

谢谢!