예제 #1
0
파일: orbm.py 프로젝트: MarcCote/iRBM
    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")
예제 #2
0
파일: orbm.py 프로젝트: orapradeep/iRBM
    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")