仅用一行代码就能可视化人工神经网络(ANNs)!

fengzhimohan 2018-04-16

仅用一行代码就能可视化人工神经网络(ANNs)!

ANN-可视化

ANN Visualizer是一个python库,它使我们能够使用一行代码就可视化人工神经网络。它和Keras一起工作,并利用python的graphviz库创建一个整洁和可视化的神经网络图。

借助先进的深度学习,您现在可以直观地看到整个深度学习过程,或可以构建卷积神经网络。

本文中,我们将使用keras构建简单的神经网络,然后使用ANN visualizer来可视化我们的神经网络。

Jason Brownlee在http://machinelearningmastery.com/上做了一些很棒的工作。他的一个例子是使用Keras来构建简单的神经网络。

安装

本演示需要3个库

  • keras

  • ANNvisualizer

  • graphviz

您可以使用以下命令安装库:

仅用一行代码就能可视化人工神经网络(ANNs)!

创建一个名为index.py的新文件,将代码复制并粘贴到文件中

仅用一行代码就能可视化人工神经网络(ANNs)!

要生成可视化文件,您需要遵循以下命令结构:

ann_vizmodelview = Truefilename =“network.gv”title =“MyNeural Network”

  • 模型—您的Keras顺序模型

  • 视图—如果设置为true,则会在执行命令后打开图形预览

  • 文件名—保存图形的位置(它以'.gv'文件格式保存)

  • 标题—可视化的ANN的标题

你刚刚看到你如何轻松地在Keras中创建你的第一个神经网络模型。

让我们将它与ann_viz()一起放入此代码中。

仅用一行代码就能可视化人工神经网络(ANNs)!

使用以下命令运行index.py:

仅用一行代码就能可视化人工神经网络(ANNs)!

以下是最终可视化内容的示例:

仅用一行代码就能可视化人工神经网络(ANNs)!

看起来很美丽,不是吗?你可以在这里(https://github.com/Prodicode/ann-visualizer)来查看模型创建和可视化的演示。

这在很多方面都很有用。当你想用一行代码解释你的模型的神经网络是怎样的时候,它可以用于教学目的。

相关推荐