提交 5c0b25f8 编写于 作者: G guosheng

Change random to np.random to be consistent between py2 and py3 in Transformer

上级 504c7f15
import glob import glob
import os import os
import random
import tarfile import tarfile
import numpy as np
class SortType(object): class SortType(object):
GLOBAL = 'global' GLOBAL = 'global'
...@@ -203,7 +204,8 @@ class DataReader(object): ...@@ -203,7 +204,8 @@ class DataReader(object):
self._token_delimiter = token_delimiter self._token_delimiter = token_delimiter
self.load_src_trg_ids(end_mark, fpattern, start_mark, tar_fname, self.load_src_trg_ids(end_mark, fpattern, start_mark, tar_fname,
unk_mark) unk_mark)
self._random = random.Random(x=seed) self._random = np.random
self._random.seed(seed)
def load_src_trg_ids(self, end_mark, fpattern, start_mark, tar_fname, def load_src_trg_ids(self, end_mark, fpattern, start_mark, tar_fname,
unk_mark): unk_mark):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册