![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
实体关系联合抽取:CasRel模型小记
文章平均质量分 64
爱喝喜茶爱吃烤冷面的小黑黑
这个作者很懒,什么都没留下…
展开
-
CasRel模型预测部分demo(这里以单个sub为例,如果为多个sub的话就遍历每个sub)
导入数据,取第39个数据data = MyDataset(path, config)dataloader = DataLoader(data, batch_size=1, shuffle=False, collate_fn=collate_fn)sample,sub_start,sub_end,relation_start,relation_end,sub_start_single,sub_end_single = data[39]text = data.tokenizer.convert_ids_原创 2021-09-12 22:34:43 · 470 阅读 · 2 评论 -
CasRel模型(联合抽取)的输入处理
数据处理输入:{“text”: “产后抑郁症@区分产后抑郁症与轻度情绪失调(产后忧郁或“婴儿忧郁”)是重要的,因为轻度情绪失调不需要治疗。”, “spo_list”: [{“Combined”: false, “predicate”: “鉴别诊断”, “subject”: “产后抑郁症”, “subject_type”: “疾病”, “object”: {"@value": “轻度情绪失调”}, “object_type”: {"@value": “疾病”}}]}输出:在这里不考虑batch_siz原创 2021-09-11 16:03:50 · 1356 阅读 · 2 评论