view sandbox/simple_autoassociator/parameters.py @ 448:0961d4b56ec5

Added some documentation
author Joseph Turian <turian@gmail.com>
date Wed, 03 Sep 2008 17:08:54 -0400
parents 4f61201fa9a9
children
line wrap: on
line source

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

import numpy

class Parameters:
    """
    Parameters used by the L{Model}.
    """
    def __init__(self, input_dimension, hidden_dimension, randomly_initialize, random_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(random_seed)
            self.w1 = (numpy.random.rand(input_dimension, hidden_dimension)-0.5)/input_dimension
            self.w2 = (numpy.random.rand(hidden_dimension, input_dimension)-0.5)/hidden_dimension
            self.b1 = numpy.zeros(hidden_dimension)
            self.b2 = numpy.zeros(input_dimension)
            #self.b2 = numpy.array([10, 0, 0, -10])
        else:
            self.w1 = numpy.zeros((input_dimension, hidden_dimension))
            self.w2 = numpy.zeros((hidden_dimension, input_dimension))
            self.b1 = numpy.zeros(hidden_dimension)
            self.b2 = numpy.zeros(input_dimension)

    def __str__(self):
        s = ""
        s += "w1: %s\n" % self.w1
        s += "b1: %s\n" % self.b1
        s += "w2: %s\n" % self.w2
        s += "b2: %s\n" % self.b2
        return s