该项目是一个简单的 LaTeX 脚本,旨在生成多层感知机(MLP)结构的神经网络图。用户只需定义每一层的神经元数量,脚本即可自动绘制出完整的网络结构图,包括输入层、隐藏层和输出层。
GitHub项目—— https://github.com/WilliamPsc/Neural-Network-Diagram-Generator
通过修改\networkShape
变量来设置每层的神经元数量。例如:
\def\networkShape{4,5,5,3}
这表示一个包含4个输入神经元、两个各5个神经元的隐藏层和3个输出神经元的网络。
\positiveWeights
、\negativeWeights
和\falseNegativeWeights
变量,指定各权重的正负情况。未列入的权重将被分类为“无符号”。\weightValues
变量定义权重值,在图中可选择显示索引或值。\displayIndex
控制是否显示权重索引。displayBias
和\displayBiasValue
设置偏置项的显示方式。可以在脚本末尾修改图示标题。默认显示网络结构和参数数量。
使用支持Tikz的LaTeX编辑器,运行以下命令生成图示:
pdflatex neural_network.tex
定义一个网络架构为4-5-5-3的网络,指定部分权重的正负情况并显示索引:
% Define the network architecture as an array
% Each element represents the number of neurons in that layer
\def\networkShape{4,5,5,3}
% Weights signs
\def\positiveWeights{0,1,2,3,4}
\def\negativeWeights{5,6,7,8,9}
\def\falseNegativeWeights{10,11,12,13,14}
% Values of all model weights
\def\weightValues{}
% Display the index number or not (0 no / 1 yes)
\pgfmathsetmacro{\displayIndex}{1}
就可以得到示例图
https://github.com/WilliamPsc/Neural-Network-Diagram-Generator
暂无评论