view sandbox/rbm/parameters.py @ 396:e0c9357456e0

Bug fixed in RBM
author Joseph Turian <turian@gmail.com>
date Tue, 08 Jul 2008 20:37:36 -0400
parents 70019965f888
children c2e6a8fcc35e
line wrap: on
line source

"""
Parameters (weights) used by the L{Model}.
"""

import numpy
import globals

class Parameters:
    """
    Parameters used by the L{Model}.
    """
    def __init__(self, input_dimension=globals.INPUT_DIMENSION, hidden_dimension=globals.HIDDEN_DIMENSION, randomly_initialize=False, seed=globals.SEED):
        """
        Initialize L{Model} parameters.
        @param randomly_initialize: If True, then randomly initialize
        according to the given seed. If False, then just use zeroes.
        """
        if randomly_initialize:
            numpy.random.seed(seed)
            self.w = (numpy.random.rand(input_dimension, hidden_dimension)-0.5)/input_dimension
            self.b = numpy.zeros((1, hidden_dimension))
            self.c = numpy.zeros((1, input_dimension))
        else:
            self.w = numpy.zeros((input_dimension, hidden_dimension))
            self.b = numpy.zeros((1, hidden_dimension))
            self.c = numpy.zeros((1, input_dimension))

    def __str__(self):
        s = ""
        s += "w: %s\n" % self.w
        s += "b: %s\n" % self.b
        s += "c: %s\n" % self.c
        return s