예제 #1
0
def batch_mapper(finput, fref, ftarget, vocabi, bsize, maxpad, maxpart, maxtoken, minbsize):

	global use_unk

	for i_d, rd, td, mlen_i, mlen_t in batch_loader(finput, fref, ftarget, bsize, maxpad, maxpart, maxtoken, minbsize):
		rsi, extok_i = map_batch(i_d, vocabi)
		rsr, extok_r = map_batch(rd, vocabi)
		yield rsi, rsr, td, mlen_i + extok_i, mlen_t + extok_r
예제 #2
0
파일: dual.py 프로젝트: lgstd/transformer
def batch_mapper(finput, ftarget, vocabi, vocabt, bsize, maxpad, maxpart,
                 maxtoken, minbsize):

    for i_d, td, mlen_i, mlen_t in batch_loader(finput, ftarget, bsize, maxpad,
                                                maxpart, maxtoken, minbsize):
        rsi, extok_i = map_batch(i_d, vocabi)
        rst, extok_t = map_batch(td, vocabt)
        yield rsi, rst, mlen_i + extok_i, mlen_t + extok_t
예제 #3
0
def batch_mapper(finput, fmt, ftarget, vocabi, vocabt, bsize, maxpad, maxpart, maxtoken, minbsize):

	global use_unk

	for i_d, md, td, mlen_i, mlen_m, mlen_t in batch_loader(finput, fmt, ftarget, bsize, maxpad, maxpart, maxtoken, minbsize):
		rsi, extok_i = map_batch(i_d, vocabi)
		rsm, extok_m = map_batch(md, vocabt)
		rst, extok_t = map_batch(td, vocabt)
		yield rsi, rsm, rst, mlen_i + extok_i, mlen_m + extok_m, mlen_t + extok_t
예제 #4
0
def batch_mapper(finput, vocabi, vocabtask, bsize, maxpad, maxpart, maxtoken,
                 minbsize):

    for i_d, taskd, mlen_i in batch_loader(finput, bsize, maxpad, maxpart,
                                           maxtoken, minbsize):
        rsi, extok_i = map_batch(i_d, vocabi)
        yield rsi, vocabtask[taskd], mlen_i + extok_i
예제 #5
0
def batch_mapper(finput, vocabi, bsize, maxpad, maxpart, maxtoken, minbsize):

	for i_d, mlen_i, nsent in batch_loader(finput, bsize, maxpad, maxpart, maxtoken, minbsize):
		rsi, extok_i = map_batch(i_d, vocabi)
		yield rsi, mlen_i + extok_i, nsent