lstm.py 文件源码

def init_params(config):
    """
    parameters for different layers including embedding, lstm, and  classifier.
    """
    params = OrderedDict()
    # embedding layer parameter
    randn = np.random.rand(config.n_words, config.dim_proj)
    params['Wemb'] = (0.01 * randn).astype(T_config.floatX)

    # lstm layer parameters
    params = param_init_lstm(config, params, prefix='lstm')

    # classifier parameters
    params['U'] = 0.01 * np.random.randn(config.dim_proj,
                                         config.ydim).astype(T_config.floatX)
    params['b'] = np.zeros((config.ydim,)).astype(T_config.floatX)

    return params