Как изобразить вес полностью связанного слоя?

Question

Я использую полностью подключенную сеть с 4 входными и 2 выходными узлами. Я сохраняю веса моей сети после того, как полностью обучу ее. Предположим, здесь моя весовая матрица

`W = np.array([[0.8,0.02],[0.5,0.4],[0.3,0.2],[0.1,0.7]])`

Я хочу визуализировать, какой вес принимает каждый класс. Как я могу это сделать? Я искал коды, связанные с этим, они используют plt.imshow. Должен ли я просто упомянуть plt.imshow (W) для визуализации весов?

Answer 1

Вы должны использовать TensorBoard для этого. Кроме того, вам не нужно хранить веса вручную, так как они хранятся в TensorFlow. Вы можете получить к ним доступ несколькими различными способами, например, с помощью tf.trainable_variables() или tape.watched_variables() в активном режиме. Тогда нужно просто перебрать переменные для нужных вам весов.

Чтобы отобразить ваши веса в TensorBoard, проверьте это: https://www.tensorflow.org/api_docs/python/tf/contrib/summary