python画神经网络模型图

Python有多个库可以用来绘制神经网络模型图，其中最常用的有matplotlib、graphviz和pydot等，以下是使用matplotlib的示例代码：

import matplotlib.pyplot as plt
import numpy as np

input_size = 784
hidden_size = 100
output_size = 10

def plot_network():
    # 创建一个新的图像
    fig, ax = plt.subplots()

    # 隐藏块中的节点
    hidden_layer = np.ones((hidden_size))
    # 输出块中的节点
    output_layer = np.ones((output_size))

    # 绘制输入层到隐藏层的连接线
    for i in range(input_size):
        for j in range(hidden_size):
            ax.plot([0, 1], [i, j + input_size], 'b', alpha=0.1)

    # 绘制隐藏层节点
    for i in range(hidden_size):
        ax.scatter([1], [i + input_size], c='r')

    # 绘制隐藏层到输出层的连接线
    for i in range(hidden_size):
        for j in range(output_size):
            ax.plot([1, 2], [i + input_size, j + input_size + hidden_size], 'g', alpha=0.1)

    # 绘制输出层节点
    for i in range(output_size):
        ax.scatter([2], [i + input_size + hidden_size], c='b')

    # 设置图像属性
    ax.set_xlim(-0.5, 2.5)
    ax.set_ylim(-0.5, input_size + hidden_size + output_size - 0.5)
    ax.axis('off')
    plt.show()

plot_network()

这个代码会生成一个简单的三层神经网络模型图，其中输入层有784个节点，隐藏层有100个节点，输出层有10个节点。你可以根据自己的需要进行更改。