僅用一行程式碼就能視覺化人工神經網路(ANNs)!

格伯納發表於2018-04-28

僅用一行程式碼就能視覺化人工神經網路(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)來檢視模型建立和視覺化的演示。

這在很多方面都很有用。當你想用一行程式碼解釋你的模型的神經網路是怎樣的時候,它可以用於教學目的。

來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/31509949/viewspace-2153491/,如需轉載,請註明出處,否則將追究法律責任。

相關文章