提交 87473db5 编写于 作者: D Deng

remove predict labels of "CLS" and "SEP"

上级 6710ab3e
...@@ -107,12 +107,8 @@ if __name__ == '__main__': ...@@ -107,12 +107,8 @@ if __name__ == '__main__':
labels = np_labels.reshape([-1]).astype(np.int32).tolist() labels = np_labels.reshape([-1]).astype(np.int32).tolist()
label_str = "" label_str = ""
count = 0 for i in range(1, np_lens-1):
for label_val in labels: label_str += inv_label_map[labels[i]]
label_str += inv_label_map[label_val]
count += 1
if count == np_lens:
break
print("%s\tpredict=%s" % (test_examples[index], label_str)) print("%s\tpredict=%s" % (test_examples[index], label_str))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册