从pdf简历中提取信息——BiLSTM-CRF

本文介绍了参加天池pdf简历信息提取比赛的经历,利用BiLSTM-CRF模型,从PDF中抽取姓名、籍贯等信息,模型在线得分0.727,排名21/1200+。通过pdfminer提取文本,BIO标注后,使用pytorch实现的BiLSTM-CRF进行训练。代码已开源,但未利用外部文本训练语言模型和词嵌入。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

0. 摘要

参加了天池的一个pdf简历信息提取的比赛,这里进行回顾、整理和分享

赛题要求从pdf简历中提取出信息,比如说名字,籍贯等。这里搭建了一个BiLSTM-CRF模型,能够从PDF简历中提取出所需的信息。

模型的线上得分是0.727,排名 21/1200+

1. 赛题相关

模型目标:pdf简历 --> 类别信息

2. 思路

在这里插入图片描述
使用python库pdfminer,将pdf简历中的文本提取出来。利用json标注文件,对提取出来的文本进行匹配和BIO标注,每一个字对应一个标注。最后,将标注后的文本送到BiLSM-CRF模型中进行训练。

3. BiLSTM-CRF 模型

在这里插入图片描述
将文本中的每个字进行one-hot编码,经过Embedding层后,每一个字对应一个字向量,所以文本可以用一个矩阵表示。将文本矩阵输入BiLSTM层,输出中每一个字会对应一个类别概率向量,此类别概率向量表示了该字属于各个类别的概率。所以所有字属于各个类别的概

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值