機械学習ではフォーマットの標準化が進んでいます。これまでTensorflowやCaffe、Chainerなど様々なソフトウェアがありましたが、生成されるモデルがONNXというフォーマットに統一されつつあります(他にもあるようですが)。 そんなONNXフォーマットのモデルをビジュアル化できるソフトウェアがNetronです。

Netronの使い方

ドロップしたところです。これは4か9を判別するモデルです。

拡大もできます。

パラメータも閲覧できます。

Netronを使うことでPythonのコードではなく、ビジュアル化されたモデルを通じて機械学習のネットワークを学ぶことができます。各ノードにおける出力の数も出ており、ネットワークの詳細が分かりやすくなるでしょう。

NetronはJavaScript製のオープンソース・ソフトウェア(MIT License)です。

Netron lutzroeder/netron: Visualizer for deep learning and machine learning models