CasRel-pytorch-reimplement icon indicating copy to clipboard operation
CasRel-pytorch-reimplement copied to clipboard
verified publisher icon longlongman

计算subject向量均值的问题

Open onelinesuhang opened this issue 4 years ago • 2 comments

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

onelinesuhang avatar Dec 02 '21 15:12 onelinesuhang

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

longlongman avatar Dec 03 '21 16:12 longlongman

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

onelinesuhang avatar Dec 04 '21 01:12 onelinesuhang