如何将模型/函数中使用的变量传递给TensorFlowEstimator

Question

在使用TensorFlowEstimator进行拟合时，如何在模型拟合后获取变量（例如嵌入表，RNN变量等），例如在skflow example中？由于tf.all_variables（）返回空列表。

Answer 1

您可以在估算工具中使用get_variable_names()。

基本上，您需要获取图表，然后调用all_variables()。例如： with self._graph.as_default(): print([v.name for v in variables.all_variables()])