前面几节课中,我们完成了模型结构和损失函数的定义,理论上说已经可以开始训练了。但这个任务的训练难度大,损失值的大小不足以说明模型好坏。要预测出完整的三元组,才算正确,所以接下来,我们需要自定义一个评估函数,来解析预测值对应的关系三元组,判断真实的准确率。

代码示例

1、调用评估函数

内容不可见,请联系管理员开通权限。

2、统计评估指标

内容不可见,请联系管理员开通权限。

3、计算关系三元组

内容不可见,请联系管理员开通权限。

到目前为止,我们训练的逻辑就全部写完了。但在训练过程中,没有做验证,因为通过训练集的准确率,也能看出模型的效果,而且这个任务也不太可能出现过拟合的情况,所以课上就不加验证逻辑了,当做一个小作业,留给大家课后自己完成。

另外,这个模型计算量很大,CPU很难出效果,所以下节课,带大家把模型放到 Kaggle 上,用它的免费 GPU 资源进行训练。

本文链接:http://ichenhua.cn/edu/note/486

版权声明:本文为「陈华编程」原创课程讲义,请给与知识创作者起码的尊重,未经许可不得传播或转售!