utils.py 文件源码

def getDataSim2(batch, nout):
    g1, g1_pos = [], []
    g2, g2_pos = [], []
    for i in batch:
        g1.append(i[0].embeddings)
        g2.append(i[1].embeddings)
        g1_pos.append(i[0].pos_embeddings)
        g2_pos.append(i[1].pos_embeddings)

    g1x, g1x_pos, g1x_mask = prepare_data(g1, g1_pos)
    g2x, g2x_pos, g2x_mask = prepare_data(g2, g2_pos)

    scores = []
    for i in batch:
        scores.append(float(i[2]))
    scores = np.asarray(scores, dtype=config.floatX)
    return (scores, g1x, g1x_pos, g1x_mask, g2x, g2x_pos, g2x_mask)