Hinton图

Hinton图对于可视化2D阵列的值（例如，权重矩阵）是有用的：正值和负值分别由白色和黑色方块表示，并且每个方块的大小表示每个值的大小。

David Warde-Farley在SciPy Cookbook上的初步想法

Hinton图示例

  1. import numpy as np
  2. import matplotlib.pyplot as plt
  5. def hinton(matrix, max_weight=None, ax=None):
  6.     """Draw Hinton diagram for visualizing a weight matrix."""
  7.     ax = ax if ax is not None else plt.gca()
  9.     if not max_weight:
  10.         max_weight = 2 ** np.ceil(np.log(np.abs(matrix).max()) / np.log(2))
  12.     ax.patch.set_facecolor('gray')
  13.     ax.set_aspect('equal', 'box')
  14.     ax.xaxis.set_major_locator(plt.NullLocator())
  15.     ax.yaxis.set_major_locator(plt.NullLocator())
  17.     for (x, y), w in np.ndenumerate(matrix):
  18.         color = 'white' if w > 0 else 'black'
  19.         size = np.sqrt(np.abs(w) / max_weight)
  20.         rect = plt.Rectangle([x - size / 2, y - size / 2], size, size,
  21.                              facecolor=color, edgecolor=color)
  22.         ax.add_patch(rect)
  24.     ax.autoscale_view()
  25.     ax.invert_yaxis()
  28. if __name__ == '__main__':
  29.     # Fixing random state for reproducibility
  30.     np.random.seed(19680801)
  32.     hinton(np.random.rand(20, 20) - 0.5)
  33.     plt.show()

下载这个示例