未验证 提交 756c09ef 编写于 作者: Y Yibing Liu 提交者: GitHub

Update reader.py (#2883)

上级 87d15f1b
...@@ -188,11 +188,11 @@ def make_one_batch_input(data_batches, index): ...@@ -188,11 +188,11 @@ def make_one_batch_input(data_batches, index):
feeding dictionary feeding dictionary
""" """
turns = np.array(data_batches["turns"][index]) turns = np.array(data_batches["turns"][index]).astype('int64')
tt_turns_len = np.array(data_batches["tt_turns_len"][index]) tt_turns_len = np.array(data_batches["tt_turns_len"][index]).astype('int64')
every_turn_len = np.array(data_batches["every_turn_len"][index]) every_turn_len = np.array(data_batches["every_turn_len"][index]).astype('int64')
response = np.array(data_batches["response"][index]) response = np.array(data_batches["response"][index]).astype('int64')
response_len = np.array(data_batches["response_len"][index]) response_len = np.array(data_batches["response_len"][index]).astype('int64')
batch_size = turns.shape[0] batch_size = turns.shape[0]
max_turn_num = turns.shape[1] max_turn_num = turns.shape[1]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册