備忘録
id:tam5917
Restricted Boltzmann Machine (RBM) をPyTorchで実装してみた
はじめに 先日、RBMの記事を見かけた。 zenn.dev RBM自体は昔からある有名なモデルであり実装の経験もあったのだが、PyTorchでキレイに書いたことはなかったので、書いてみたということ。 実装 以下に置いた。Enjoy! train_rbm.py · GitHub train_rbm2.py · GitHub 今回はデモンストレーションということで、MNIST上で訓練…