Information-Extraction-Chinese icon indicating copy to clipboard operation
Information-Extraction-Chinese copied to clipboard

模型效果问题

Open hddy2000 opened this issue 6 years ago • 6 comments

您好,我根据您的项目复盘一下模型训练,但发现同样训练9000轮后模型的效果和您那个model-9000的效果差很多。 我的结果: ['大黄', '小黄', '大黄和小黄一起玩了游戏'] 实体1: 大黄 实体2: 小黄 大黄和小黄一起玩了游戏 关系是: No.1: 父母, Probability is 0.9015224 No.2: 夫妻, Probability is 0.082126126 No.3: 上下级, Probability is 0.006153937 您的结果: ['大黄', '小黄', '大黄和小黄一起玩了游戏'] 实体1: 大黄 实体2: 小黄 大黄和小黄一起玩了游戏 关系是: No.1: 合作, Probability is 0.73008347 No.2: 好友, Probability is 0.19676886 No.3: 情侣, Probability is 0.05344718

很想知道造成的原因是什么? 您的那个9000 的模型是当 soft_max_loss下降到一定程度就停止训练了吗?还是说直接设置run了9000个step?训练停止的条件是什么? 还请不吝赐教。

hddy2000 avatar Dec 12 '18 09:12 hddy2000

补充一下,我在你那个test测试集上跑了一下,你那个模型的PR曲线面积是0.43,我的PR曲线面积只有0.18.。。不知道是什么原因。。。

hddy2000 avatar Dec 14 '18 03:12 hddy2000

求问,为什么我的step = 150 就停止了。。

AntonZhao avatar Jan 10 '19 03:01 AntonZhao

@AntonZhao 你可以查看代码里面的参数设置

WeiYangBin avatar Dec 18 '19 08:12 WeiYangBin

@hddy2000 数据集不一样所以test PR曲线的面积也不一样

WeiYangBin avatar Dec 23 '19 01:12 WeiYangBin

我是句子长度设置的比较大,为512,并且句子包含多对实体,跑出来的结果就不是很准确。

s1162276945 avatar Feb 06 '20 06:02 s1162276945

我觉得模型预测的结果比你给的结果看起来更好

logiccao avatar Sep 14 '20 07:09 logiccao