skyfaker / chinese_address_extract

bert+biafine模型和bert+bi-lstm+crf模型抽取中文地址中的实体

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

介绍

使用bert+biafine模型和bert+bi-lstm+crf模型抽取中文地址中的实体(具体类别见train_data/中文地址要素解析标注规范.pdf

文件结构

code: 代码文件

main_biafine.py: 基于bert和双仿射的模型入口
dataset: 数据类定义
model: 模型和网络定义
utils: 工具方法

train_data: 数据文件,包括数据分析、转换和生成代码

使用方法

使用bert_biaffine模型:执行python main_biafine.py 即可

使用bert_bi-lstm+crf模型:按照main_biafine.py文件替换成对应的网络即可

About

bert+biafine模型和bert+bi-lstm+crf模型抽取中文地址中的实体


Languages

Language:Python 100.0%