fix bug (multi-label reader)

35dc0571 · zhangxuefei · fa68672a · 35dc0571
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddlehub/reader/nlp_reader.py paddlehub/reader/nlp_reader.py +1 -1

未找到文件。
--- a/paddlehub/reader/nlp_reader.py
+++ b/paddlehub/reader/nlp_reader.py
@@ -696,7 +696,7 @@ class MultiLabelClassifyReader(BaseReader):
            label_ids = [0, 0, 0, 0, 0, 0]
        else:
            for label in example.label:
-                label_ids.append(self.label_map[label])
+                label_ids.append(int(label))

        if phase != "predict":
            Record = namedtuple(