search

tonytan48 / KD-DocRE

Implementation of Document-level Relation Extraction with Knowledge Distillation and Adaptive Focal Loss
110 stars 20 forks source link

Sample code to run inference on any text #15

Open AlexisNvn opened 2 years ago

AlexisNvn commented 2 years ago

Good evening, Would you have any advices/indications on how to create a simple script that compute the model outputs given any text in Re-DocRED format ?

Thanks & keep going with the great work ! Alexis

tonytan48 commented 2 years ago

Hi @AlexisNvn, thank you for your interest in our work! Unfortunately, our model cannot be used to directly infer on any raw text. This is because our model is only classifying the relations. We need an additional NER model to extract the entity mentions and group them together (coreference resolution). I can try look for some NER and entity linking model and see if we can achieve that function.