Exemplo n.º 1
0
    def __init__(self,
                 input_size,
                 hidden_size,
                 beta=1,
                 penalty="softplus_bi",
                 *args, **kwargs):

        RBM.__init__(self, input_size, hidden_size, *args, **kwargs)

        self.penalty = penalty
        self.beta = theano.shared(np.array(beta, dtype=theano.config.floatX), name="beta")
Exemplo n.º 2
0
    def __init__(self,
                 input_size,
                 hidden_size,
                 beta=1,
                 penalty="softplus_bi",
                 *args,
                 **kwargs):

        RBM.__init__(self, input_size, hidden_size, *args, **kwargs)

        self.penalty = penalty
        self.beta = theano.shared(np.array(beta, dtype=theano.config.floatX),
                                  name="beta")