view sandbox/simple_autoassociator/parameters.py @ 404:8cc11ac97087

Debugging simple AA a bit
author Joseph Turian <turian@gmail.com>
date Thu, 10 Jul 2008 00:51:32 -0400
parents 36baeb7125a4
children faffaae0d2f9
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.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