Question

如何从element.getAttribute("attribute name")中选择一个图层并访问该图层中每个单位的权重向量？具体来说，我正在尝试可视化密集层的权重。

查看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 has a method调用了join。因此，一旦您生成了一个检查点（或从一个加载了变量值），并且如果您知道密集层的名称，则可以使用matplotlib执行类似的操作：

get_variable_value

Answer 2

我只是使用预编译的Estimator进行测试，因此对我来说工作正常。

import matplotlib.pyplot as plt

names = classifier.get_variable_names()        
print("name:", names)
for i in names:
    print(classifier.get_variable_value(i)

如何可视化TensorFlow Estimator权重？

2 个答案: