如何可視化TensorFlow估計器權重？

Question

如何從tf.estimator.Estimator中選擇一個圖層並訪問該圖層中每個單元的權向量？具體來說，我試圖想象密集層的權重。

看着https://github.com/tensorflow/tensorflow/blob/r1.3/tensorflow/python/layers/core.py似乎權被稱爲內核，但我不能使用估算抽象時訪問這些。

PS：對於估計的實現的一個例子，讓我們引用https://www.tensorflow.org/get_started/estimator

Answer 1

估計has a method稱爲get_variable_value。所以，一旦你獲得了一個檢查點（或從一個加載的變量值），如果你知道的緻密層的名稱，你可以使用matplotlib做這樣的事情：

import matplotlib.pyplot as plt 

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