pyunit-ner 2021.8.2

百度实体抽取模型

pyUnit-NER

NER模块集合

安装

pip install pyunit-ner

推荐使用Docker部署

docker pull jtyoui/pyunit-ner
docker run -d -P jtyoui/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())

抽取实体接口文档

http://ip:port/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"
}

---