Fine-tuning transformers for relation classification

Beginners
1

Hi all,

I am working on a relation classification task and I already have a training set with highlighted entities. After doing some research, it seems that ERNIE and ERNIE 2.0 would be good candidates for fine-tuning. They are also kindly provided in the model hub by nghuyong!

Would you suggest any other models for a relation classification task? Any other tips are welcome too :slight_smile:

I’m adding below an example of how the text data look like:

'[HT]John Doe[HT] was an important contributor to our [TL]software project[TL].'

And the classes are Positive, Negative, Unknown.

Thank you!