多使用namedtuple,这样可以在返回一个数据的时候还能带有相关信息
Document = namedtuple('Document',
'content is_validation is_test label add_tokens')
yield Document(
content=content,
is_validation=is_validation,
is_test=False,
label=class_label,
add_tokens=True)
使用的时候就可以直接使用这个属性了。
doc.label
网友评论