python画神经网络结构图_画出神经网络结构图

学习资料:

使用 Viznet 画出神经网络结构图

'''

'''

import numpy as np

from viznet import connecta2a, node_sequence, NodeBrush, EdgeBrush, DynamicShow

def draw_feed_forward(ax, num_node_list):

'''

draw a feed forward neural network.

Args:

num_node_list (list): 每层节点数组成的列表

'''

num_hidden_layer = len(num_node_list) - 2 # 隐藏层数

token_list = ['\sigma^z'] + \

['y^{(%s)}' % (i + 1) for i in range(num_hidden_layer)] + ['\psi']

kind_list = ['nn.input'] + ['nn.hidden'] * num_hidden_layer + ['nn.output']

radius_list = [0.3] + [0.2] * num_hidden_layer + [0.3] # 半径大小

y_list = - 1.5 * np.arange(len(num_node_list)) # 每一层节点所在的位置的纵轴坐标，全取负值说明网络是自顶而下的

seq_list = []

for