上传者: 42097967
|
上传时间: 2021-10-12 20:31:39
|
文件大小: 10.96MB
|
文件类型: -
概述
这是用Python编写的 RBM 学习算法的简单实现,它使用numpy包来加速矩阵计算。
路线
要使用默认参数在 MNIST 数据集上运行 RBM 学习算法,请键入
python rbm.py
否则您可以自己指定所有参数
python rbm.py
这将在训练示例上训练 RBM,然后尝试在给定学习权重的情况下重建测试图像。 原始测试图像和重建的测试图像都将保存在Output/文件夹中。
依赖关系
请注意,此项目需要 Python 3.x 才能运行。 它还需要numpy包。