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
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
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
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
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