仅用一行代码就能可视化人工神经网络(ANNs)!
ANN Visualizer是一个python库,它使我们能够使用一行代码就可视化人工神经网络。它和Keras一起工作,并利用python的graphviz库创建一个整洁和可视化的神经网络图。
借助先进的深度学习,您现在可以直观地看到整个深度学习过程,或可以构建卷积神经网络。
本文中,我们将使用keras构建简单的神经网络,然后使用ANN visualizer来可视化我们的神经网络。
Jason Brownlee在http://machinelearningmastery.com/上做了一些很棒的工作。他的一个例子是使用Keras来构建简单的神经网络。
安装
本演示需要3个库
keras
ANNvisualizer
graphviz
您可以使用以下命令安装库:
创建一个名为index.py的新文件,将代码复制并粘贴到文件中
要生成可视化文件,您需要遵循以下命令结构:
ann_viz(model,view = True,filename =“network.gv”,title =“MyNeural Network”)
模型—您的Keras顺序模型
视图—如果设置为true,则会在执行命令后打开图形预览
文件名—保存图形的位置(它以'.gv'文件格式保存)
标题—可视化的ANN的标题
你刚刚看到你如何轻松地在Keras中创建你的第一个神经网络模型。
让我们将它与ann_viz()一起放入此代码中。
使用以下命令运行index.py:
以下是最终可视化内容的示例:
看起来很美丽,不是吗?你可以在这里(https://github.com/Prodicode/ann-visualizer)来查看模型创建和可视化的演示。
这在很多方面都很有用。当你想用一行代码解释你的模型的神经网络是怎样的时候,它可以用于教学目的。