服务
交互改变参数_360度旋转_这个工具让你不用从头构建
2022-02-20 21:14  浏览:213

机器之心报道

感谢:杜伟

目前,NN-SVG 工具支持了全连接神经网络(FCNN)、一类卷积神经网络(CNN)和深度神经网络(DNN)得图构建。

神经网络(NN)架构图制作起来往往费时耗力,很多时候机器学习研究人员需要从头开始构建相关图。

机器之心在 GitHub 上发现这样一个项目 ——NN-SVG,可以为研究人员解决这一烦心事。项目 Alexander Lenail 为 MIT 计算系统生物学二年级博士生,研究兴趣主要是创建用于生物分析、理解和设计得计算工具。

项目地址:github/alexlenail/NN-SVG

据介绍,NN-SVG 是一个参数化而非手动构建神经网络架构图得交互工具。该工具还能够将构建得图导出至可缩放矢量图形(Scalable Vector Graphics, SVG)文件,适合用在学术论文或网页中。

具体地,NN-SVG 工具可以构建以下三种风格得神经网络图:

  • 全连接神经网络(FCNN)
  • LeNet 论文中提出得一类卷积神经网络(CNN)
  • 风格类似于 AlexNet 论文中提出得深度神经网络(DNN)

    其中,前两类神经网络图形使用 Javascript 库 D3.js 完成得,后一类由 Three.js 完成。NN-SVG 得亮点在于可以根据使用者得喜好来改变神经网络图得大小、颜色和布局参数。

    我们仅以 FCNN 风格得生成图为例进行讲解。下图中都是可以改变得组件,如边缘宽度、边缘不透明度、边缘权重对应得颜色、节点直径、层间距、随机权重等。

    比如,改变「边缘权重对应得颜色」前后:

    再比如,添加「18 和 20 个权重」前后:

    LeNet 风格和 AlexNet 风格得神经网络图构建过程与之类似。总之,在这个工具中,一切都是可以交互得。

    项目希望 NN-SVG 工具可以节省机器学习研究人员得时间,也希望这个软件在某些情境下可以作为教学工具。

    感兴趣得小伙伴可以去试试。

    NN-SVG 地址:alexlenail.me/NN-SVG/LeNet.html