neuro.py 文件源码

def __init__(self, quadratic_filters_ex=2, quadratic_filters_sup=2, W_quad_ex_initializer='glorot_uniform',
                 W_quad_sup_initializer='glorot_uniform', W_lin_initializer='glorot_uniform',
                 W_quad_ex_regularizer=None, W_quad_sup_regularizer=None, W_lin_regularizer=None,
                 W_quad_ex_constraint=None, W_quad_sup_constraint=None, W_lin_constraint=None,
                 **kwargs):

        self.quadratic_filters_ex = quadratic_filters_ex
        self.quadratic_filters_sup = quadratic_filters_sup

        self.W_quad_ex_initializer = initializers.get(W_quad_ex_initializer)
        self.W_quad_sup_initializer = initializers.get(W_quad_sup_initializer)
        self.W_lin_initializer = initializers.get(W_lin_initializer)

        self.W_quad_ex_constraint = constraints.get(W_quad_ex_constraint)
        self.W_quad_sup_constraint = constraints.get(W_quad_sup_constraint)
        self.W_lin_constraint = constraints.get(W_lin_constraint)

        self.W_quad_ex_regularizer = regularizers.get(W_quad_ex_regularizer)
        self.W_quad_sup_regularizer = regularizers.get(W_quad_sup_regularizer)
        self.W_lin_regularizer = regularizers.get(W_lin_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(RustSTC, self).__init__(**kwargs)