机器学习----使用python仿写theano中RBM的源代码

置顶

liugan5371375

于 2016-04-29 09:33:17 发布

阅读量2.7k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：机器学习文章标签： RBM源代码

本文链接：https://blog.youkuaiyun.com/liugan5371375/article/details/51279373

本文介绍了如何使用Python语言模仿Theano库中的受限玻尔兹曼机（RBM）源代码，重点在于构建激活函数，为RBM的实现提供关键支持。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# -*- coding: utf-8 -*-
'''
Created on 2016年4月1日

@author: LIU
'''
import sys
import numpy
import matplotlib.pylab as plt
import numpy as np
import random
from scipy.linalg import norm
import PIL.Image
from utils import *

class RBM(object):
    def __init__(self, input=None, n_visible=2, n_hidden=3, \
        W=None, hbias=None, vbias=None, rng=None):

        self.n_visible = n_visible  # num of units in visible (input) layer
        self.n_hidden = n_hidden    # num of units in hidden layer

        if rng is None:
            rng = numpy.random.RandomState(1234)


        if W is None:
            a = 1. / n_visible
            initial_W = numpy.array(rng.uniform(  # initialize W uniformly(随机生成实数在-a-a之间)
                low=-a,
                high=a,
                size=(n_visible, n_hidden)))

            W = initial_W

        if hbias is None:
            hbias = numpy.zeros(n_hidden)  # initialize h bias 0

        if vbias is None:
            vbias = numpy.zeros(n_visible)  # initialize v bias 0