search

longlongman / CasRel-pytorch-reimplement

Pytorch reimplement of the paper "A Novel Cascade Binary Tagging Framework for Relational Triple Extraction" ACL2020. The original code is written in keras.
197 stars 52 forks source link

计算subject向量均值的问题 #20

Open onelinesuhang opened 2 years ago

onelinesuhang commented 2 years ago

casrel.py 中 sub = (sub_head + sub_tail) / 2，是不是只计算了subject头和尾词的均值，原论文使用的是对整个span计算均值，按照您的代码，如果subject超过两个词，是不是会出现一些误差

longlongman commented 2 years ago

我记得在原论文中，模型只是预测实体的开始位置与结束位置

onelinesuhang commented 2 years ago

模型是只预测开始和结束位置，但是论文中说的是计算这个span的均值，后来我看了论文原来的代码，好像也是开始位置和结束位置