提交 35dc0571 编写于 作者: Z zhangxuefei

fix bug (multi-label reader)

上级 fa68672a
...@@ -696,7 +696,7 @@ class MultiLabelClassifyReader(BaseReader): ...@@ -696,7 +696,7 @@ class MultiLabelClassifyReader(BaseReader):
label_ids = [0, 0, 0, 0, 0, 0] label_ids = [0, 0, 0, 0, 0, 0]
else: else:
for label in example.label: for label in example.label:
label_ids.append(self.label_map[label]) label_ids.append(int(label))
if phase != "predict": if phase != "predict":
Record = namedtuple( Record = namedtuple(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册