Python RBM.__init__の例

コード例 #1

ファイル: 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

ファイル: 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")