機械学習ではフォーマットの標準化が進んでいます。これまでTensorflowやCaffe、Chainerなど様々なソフトウェアがありましたが、生成されるモデルがONNXというフォーマットに統一されつつあります(他にもあるようですが)。
そんなONNXフォーマットのモデルをビジュアル化できるソフトウェアがNetronです。
Netronの使い方
ドロップしたところです。これは4か9を判別するモデルです。
拡大もできます。
パラメータも閲覧できます。
Netronを使うことでPythonのコードではなく、ビジュアル化されたモデルを通じて機械学習のネットワークを学ぶことができます。各ノードにおける出力の数も出ており、ネットワークの詳細が分かりやすくなるでしょう。
NetronはJavaScript製のオープンソース・ソフトウェア(MIT License)です。
lutzroeder/netron: Visualizer for deep learning and machine learning models
MOONGIFTの関連記事