如何可视化TensorFlow Estimator权重？

Question

如何从a中选择一个图层tf.estimator.Estimator并访问该图层中每个单元的权重向量？具体来说,我正在尝试可视化密集层的权重.

看看https://github.com/tensorflow/tensorflow/blob/r1.3/tensorflow/python/layers/core.py看起来权重被称为内核,但是在使用时我无法访问权限Estimator抽象.

Ps:有关Estimator实现的示例,请参考https://www.tensorflow.org/get_started/estimator

Answer 1

Estimator 有一个叫做的方法get_variable_value.所以,一旦你产生了一个检查点(或从一个加载了变量值),如果你知道密集层的名称,你可以使用matplotlib做这样的事情:

import matplotlib.pyplot as plt

weights = estimator.get_variable_value('dense/kernel')
plt.imshow(weights, cmap='gray')
plt.show()