これは、カプセル間の動的ルーティングのPyTorch実装/チュートリアルです。
カプセルネットワークは、フィーチャをカプセルとして埋め込み、投票メカニズムを使用して次のカプセル層にルーティングするニューラルネットワークアーキテクチャです。
他のモデルの実装とは異なり、モジュールだけでは一部の概念を理解するのが難しいため、サンプルを用意しています。
これは、カプセルを使用して MNIST データセットを分類するモデルの注釈付きコードです。このファイルには、Capsule Networks のコアモジュールの実装が格納されています。
Jindongwang/Pytorch-Capsulenetを使って、論文に関する混乱を解消しました。
これは、MNISTデータセットでカプセルネットワークをトレーニングするためのノートブックです。