-
Tensorflow-rbm是受限玻尔兹曼机在Tensorflow中的具体实现
资源介绍
张量流
深度玻纤编码器的分层预训练的受限玻尔兹曼机的Tensorflow实现。
这是的分支,并进行了一些更正和改进。
受限玻尔兹曼机是一种遗留机器学习模型,不再在任何地方使用。 该存储库仅具有历史和教育价值。 我已经使用更新了代码以在现代系统上运行,但是我将不再维护它。
安装
git clone https://github.com/meownoid/tensorfow-rbm.git
cd tensorfow-rbm
python -m pip install -r requirements.txt
python setup.py
例子
Bernoulli-Bernoulli RBM适用于Bernoulli分布的二进制输入数据。 例如,MNIST。 要训练模型,只需构造tf.data.Dataset其中包含形状为(n_visible,)向量(n_visible,)并将其传递给fi