Skip to content

andreYoo/pytorch-gbrbm

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pytorch-gbrbm

Gaussian Bernoulli RBM based on Pytorch Lib (Original RBM is referred from https://github.com/GabrielBianconi/pytorch-rbm)

Files. .
.
├── gbrbm_example.py
├── gbrbm.py
├── LICENSE
├── README.md
└── utils.py

How to run python gbrbm_example.py

Output

Loading dataset...
Training RBM...
Epoch Error (epoch=0): 6593016.5000
Epoch Error (epoch=1): 3434561.7500
Epoch Error (epoch=2): 2627748.5000
Epoch Error (epoch=3): 2355443.5000
Epoch Error (epoch=4): 2217778.7500
Epoch Error (epoch=5): 2132137.0000
Epoch Error (epoch=6): 2072602.3750
Epoch Error (epoch=7): 2028256.8750
Epoch Error (epoch=8): 1992789.1250
Epoch Error (epoch=9): 1962103.7500\

dependendies. Python 3.6

Multi-layered version will be updated.

About

Gaussian Bernoulli RBM based on Pytorch Lib

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages