search

WindChimeRan / pytorch_multi_head_selection_re

BERT + reproduce "Joint entity recognition and relation extraction as a multi-head selection problem" for Chinese and English IE
138 stars 31 forks source link

bio gold in not train mode #10

Closed antgr closed 4 years ago

antgr commented 4 years ago

https://github.com/WindChimeRan/pytorch_multi_head_selection_re/blob/30c8aa1b2d89a115612d2c5f344e96ab132220c8/lib/models/selection.py#L191 Hi is this the bio_gold? Or is it inferred?

WindChimeRan commented 4 years ago

The training and inference are different.

bio_gold is used for bio embedding latter.

Training: use gold BIO

Testing: use inferred BIO