未验证 提交 95f0a9de 编写于 作者: Z zzszmyf 提交者: GitHub

Update reader.py

resolve #1794
_reader_creator need default value,if not, the code can't work
上级 e3c3af79
...@@ -13,7 +13,7 @@ class CriteoDataset(Dataset): ...@@ -13,7 +13,7 @@ class CriteoDataset(Dataset):
self.continuous_range_ = range(1, 14) self.continuous_range_ = range(1, 14)
self.categorical_range_ = range(14, 40) self.categorical_range_ = range(14, 40)
def _reader_creator(self, file_list, is_train, trainer_num, trainer_id): def _reader_creator(self, file_list, is_train, trainer_num, trainer_id=0):
def reader(): def reader():
for file in file_list: for file in file_list:
with open(file, 'r') as f: with open(file, 'r') as f:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册