search

Babelscape / rebel

REBEL is a seq2seq model that simplifies Relation Extraction (EMNLP 2021).
474 stars 71 forks source link

Fine tuning for person to person entity relationship extraction #65

Open Theosagie opened 1 year ago

Theosagie commented 1 year ago

Please how cam I fine tune this model on my dataset to predict triplets of only person to person entities and their relationships. I am using CustFRE dataset 1-s2.0-S2352340922001913-mmc3.xlsx custFRE dataset sentences.xlsx CustFRE Dataset with expected relationship.xlsx relationsKey.txt Thanks

LittlePea13 commented 1 year ago

Hi @Theosagie. I am sorry but I didn't understand how the data you shared can be framed as a Relation Extraction task, could you perhaps give some example?

In any case, you can look at how REBEL was finetuned on CONLL04 or NYT to have an example on how to train it on your own dataset.