上节课,我们在 Dataset 类中,解析出了三元组的 id 信息,但是我们要输入模型进行运算的数据,不能是 id,而应该是一个序列。所以接下来,我们借助 DataLoader 的 collate_fn 中间函数,来实现 id 转序列的过程。

模型参数结构分析

batch_mask, (batch_text, batch_sub_rnd), (batch_sub, batch_obj_rel)

代码示例

1、修改 Dataloader 参数

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

2、定义 collate_fn 函数和变量初始化

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

3、位置信息转 onehot

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

4、循环解析内容

这一步用 torch.nn.utils.rnn.pad_sequence 写法上会相对简洁,但理解难度比较大,还容易出错,我们还是用最直接的循环遍历的方式处理。

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

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

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