wcirq / pyunit-ner

NER实体识别模型

pyUnit-NER

NER模块集合

安装

pip install pyunit-ner

默认官方数据集训练的模型（只能识别：人名、地名、机构名）

点击下载模型

默认的参数和映射表

import pprint
from pyunit_ner import ernie_st, ernie_match, parseNER


def test():
    # 默认的模型参数和映射表
    model = '/home/jtyoui/Documents/model'
    s = ernie_st(new_model_path=model)
    data = ernie_match('刘万光对李伟说：在贵阳市南明村永乐乡发生了一件恐怖的事情', s)
    result = parseNER(data)
    return result


if __name__ == '__main__':
    pprint.pprint(test())

官网地址

点击ERNIE查看地址

DockerHub镜像下载

docker pull jtyoui/pyunit-ner
docker run -d -P -v 下载模型的地址:/mnt/model jtyoui/pyunit-ner

抽取实体接口文档

http://xxx.xxx.xxx.xxx:端口/docs

请求报文

参数名	类型	NULL	说明
data	string	Yes	数据

请求示例

import requests

url = "http://127.0.0.1:9000/pyunit/ner?data=我在贵州贵阳观山湖"
headers = {'Content-Type': "application/x-www-form-urlencoded"}
response = requests.get(url).json()
print(response)

返回报文

参数名	类型	NULL	说明
msg	string	Yes	返回消息
data	list	Yes	标注数据类型
address	list	Yes	地址
person	list	Yes	人名
org	list	Yes	机构名

{
    "code": 200,
    "entity": {
        "address": [
            "贵州贵阳观山湖"
        ],
        "number": [
            "6",
            "6",
            "4",
            "5",
            "4",
            "5",
            "4",
            "5",
            "5"
        ],
        "organization": [],
        "person": [],
        "word": [
            "我",
            "在",
            "贵",
            "州",
            "贵",
            "阳",
            "观",
            "山",
            "湖"
        ]
    },
    "msg": "success"
}

---