lavis-nlp / spert

PyTorch code for SpERT: Span-based Entity and Relation Transformer
MIT License
685 stars 146 forks source link

Trained model : Relation classification is bad #68

Closed yy2lyx closed 1 year ago

yy2lyx commented 2 years ago

Trained with my own dataset, I see the entity and relation loss both decrease . When I evaluete trained model , the entity classfication is great, but the relation classification is realy bad.

Somebody help me ! Thanks in advance