namedtuple

2018-12-31  本文已影响0人  VanJordan

多使用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
上一篇 下一篇

猜你喜欢

热点阅读