Unable to find the entities from the predicted result

helloeve / mre-in-one-pass

Implementation for Extracting Multiple-Relations in One-Passwith Pre-Trained Transformers

Apache License 2.0

80 stars 20 forks source link

Unable to find the entities from the predicted result #3

Closed kedir closed 5 years ago

kedir commented 5 years ago

Hi, I have successfully trained the model and did prediction, but I couldn't get the entities along with the relationship predicted. I was wondering which entities from the sentence that the predicted relationship is connecting to. Thanks.

helloeve commented 5 years ago

@kedir Please refer to https://github.com/helloeve/mre-in-one-pass/blob/master/data/semeval2018/single/SemEval.train.tsv for the entity location and relation type. Each line should contain 4 numbers which refer to entity1_start_index, entity1_end_index, entity2_start_index, entity2_end_index