training.py 文件源码

def init_params(options):
    params = OrderedDict()

    # lstm params
    if options['pretrain'] == '':
        params['Wemb'] = 1.0 * numpy.random.rand(options['n_words'], options['dim_emb']).astype(config.floatX)
    else:
        file = 'best_model/' + options['disease'] + '/best.' + options['pretrain'] + '.pkl'
        f = open(file, 'rb')
        opt, model = cPickle.load(f)
        params['Wemb'] = model['Wemb']

    params = init_lstm_params(options, params)

    # top layer
    init_top_params = tasks[options['task']][4]
    params = init_top_params(options, params)
    return params