def get_batch_data(x,
y,
sen_len,
doc_len,
word_dis,
keep_prob1,
keep_prob2,
batch_size,
test=False):
if y == None:
for index in func.batch_index(len(word_dis), batch_size, test):
feed_list = [
x[index], sen_len[index], doc_len[index], word_dis[index],
keep_prob1, keep_prob2
]
yield feed_list, len(index)
else:
for index in func.batch_index(len(y), batch_size, test):
feed_list = [
x[index], y[index], sen_len[index], doc_len[index],
word_dis[index], keep_prob1, keep_prob2
]
yield feed_list, len(index)
def get_batch_data(x,
sen_len,
doc_len,
keep_prob1,
keep_prob2,
y,
batch_size,
test=False):
for index in func.batch_index(len(y), batch_size, test):
feed_list = [
x[index], sen_len[index], doc_len[index], keep_prob1, keep_prob2,
y[index]
]
yield feed_list, len(index)
def get_cause_batch_data(x,
y_cause,
sen_len,
doc_len,
clause_pos,
keep_prob1,
keep_prob2,
batch_size,
test=False):
for index in func.batch_index(len(y_cause), batch_size, test):
feed_list = [
x[index], y_cause[index], sen_len[index], doc_len[index],
clause_pos[index], keep_prob1, keep_prob2
]
yield feed_list, len(index)
def get_batch_data(x,
sen_len,
doc_len,
keep_prob1,
keep_prob2,
y_position,
y_cause,
batch_size,
test=False):
for index in batch_index(len(y_cause), batch_size, test):
feed_list = [
x[index], sen_len[index], doc_len[index], keep_prob1, keep_prob2,
y_position[index], y_cause[index]
]
yield feed_list, len(index)
