neuro.py 文件源码

def __init__(self, weights=None, kernel_initializer='glorot_uniform',
                 alpha_initializer='ones', alpha_regularizer=None, alpha_constraint=None,
                 beta_delta_initializer='ones', beta_delta_regularizer=None, beta_delta_constraint=None,
                 gamma_eta_initializer='ones', gamma_eta_regularizer=None, gamma_eta_constraint=None,
                 rho_initializer='ones', rho_regularizer=None, rho_constraint=None,
                 **kwargs):

        self.alpha_initializer = initializers.get(alpha_initializer)
        self.beta_delta_initializer = initializers.get(beta_delta_initializer)
        self.gamma_eta_initializer = initializers.get(gamma_eta_initializer)
        self.rho_initializer = initializers.get(rho_initializer)

        self.alpha_constraint = constraints.get(alpha_constraint)
        self.beta_delta_constraint = constraints.get(beta_delta_constraint)
        self.gamma_eta_constraint = constraints.get(gamma_eta_constraint)
        self.rho_constraint = constraints.get(rho_constraint)

        self.alpha_regularizer = regularizers.get(alpha_regularizer)
        self.beta_delta_regularizer = regularizers.get(beta_delta_regularizer)
        self.gamma_eta_regularizer = regularizers.get(gamma_eta_regularizer)
        self.rho_regularizer = regularizers.get(rho_regularizer)

        self.input_spec = [InputSpec(ndim=2)]

        if 'input_shape' not in kwargs and 'input_dim' in kwargs:
            kwargs['input_shape'] = (kwargs.pop('input_dim'),)
        super(NakaRushton, self).__init__(**kwargs)