search

wzhouad / ATLOP

Source code for paper "Document-Level Relation Extraction with Adaptive Thresholding and Localized Context Pooling", AAAI 2021
197 stars 39 forks source link

Inference #15

Closed CJPJ007 closed 3 years ago

CJPJ007 commented 3 years ago

Can you please tell how can I perform inference and get (entity, relation, entity) type of output?

wzhouad commented 3 years ago

Hi,

Not sure which dataset you are referring to. If it's on DocRED, the to_offcial function in evaluation.py can do.