Information-Extraction-Chinese
Information-Extraction-Chinese copied to clipboard
模型效果问题
您好,我根据您的项目复盘一下模型训练,但发现同样训练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?训练停止的条件是什么? 还请不吝赐教。
补充一下,我在你那个test测试集上跑了一下,你那个模型的PR曲线面积是0.43,我的PR曲线面积只有0.18.。。不知道是什么原因。。。
求问,为什么我的step = 150 就停止了。。
@AntonZhao 你可以查看代码里面的参数设置
@hddy2000 数据集不一样所以test PR曲线的面积也不一样
我是句子长度设置的比较大,为512,并且句子包含多对实体,跑出来的结果就不是很准确。
我觉得模型预测的结果比你给的结果看起来更好