跳到内容

alexenail公司/NN-SVG

许可证:麻省理工学院 地位|文件|贡献

制作神经网络架构的插图通常很耗时,机器学习研究人员经常发现自己从头开始构建这些图表。

NN-SVG是一种用于参数化而非手动创建神经网络(NN)架构图的工具。它还提供了将这些图形导出为可缩放矢量图形(SVG)文件的能力,这些文件适合包含在学术论文或网页中。

该工具可以生成三种图形:经典的全连接神经网络(FCNN)图形、卷积神经网络(CNN)图形LeNet论文和Deep Neural Network图形遵循中引入的样式AlexNet论文。前两个是使用D3 javascript库后者使用javascript库三个.js.NN-SVG可以通过多种尺寸、颜色和布局参数,根据用户的喜好设置图形的样式。

我希望这个工具可以节省机器学习研究人员的时间,我希望这个软件在某些情况下也可以用作教学工具。

引用

LeNail,(2019)。NN-SVG:出版物-就绪神经网络架构示意图。
开源软件杂志,4(33),747,https://doi.org/10.21105/joss.00747

相关的