ANN Visualizer是一個python庫,它使我們能夠使用一行程式碼就視覺化人工神經網路。它和Keras一起工作,並利用python的graphviz庫建立一個整潔和視覺化的神經網路圖。
藉助先進的深度學習,您現在可以直觀地看到整個深度學習過程,或可以構建卷積神經網路。
本文中,我們將使用keras構建簡單的神經網路,然後使用ANN visualizer來視覺化我們的神經網路。
Jason Brownlee在上做了一些很棒的工作。他的一個例子是使用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:
以下是最終視覺化內容的示例:
看起來很美麗,不是嗎?你可以在這裡()來檢視模型建立和視覺化的演示。
這在很多方面都很有用。當你想用一行程式碼解釋你的模型的神經網路是怎樣的時候,它可以用於教學目的。