File size: 1,286 Bytes
f22791f
 
 
 
 
 
 
 
 
 
 
 
ba9c6c6
e6f30f0
ba9c6c6
 
 
 
 
 
 
 
 
61a7c58
ba9c6c6
 
 
 
61a7c58
ba9c6c6
61a7c58
ba9c6c6
61a7c58
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ba9c6c6
61a7c58
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
---
language:
- zh
tags:
- relation extraction
license: apache-2.0
datasets:
- DuIE
metrics:
- micro f1
---

# GPLinker关系抽取模型

## 模型介绍

+ 数据集:百度 `DUIE2.0` 信息抽取

+ 模型方法:[GPLinker:基于GlobalPointer的实体关系联合抽取](https://kexue.fm/archives/8888)

## 使用方法

```commandline
pip install litie
```

```python
from pprint import pprint
from litie.pipelines import RelationExtractionPipeline

pipeline = RelationExtractionPipeline("gplinker", model_name_or_path="xusenlin/duie-gplinker", model_type="bert")
text = "查尔斯·阿兰基斯(Charles Aránguiz),1989年4月17日出生于智利圣地亚哥,智利职业足球运动员,司职中场,效力于德国足球甲级联赛勒沃库森足球俱乐部。"
pprint(pipeline(text))

# 输出
[
  {
    "出生地": [
      {
        "subject": "查尔斯·阿兰基斯",
        "object": "智利圣地亚哥"
      }
    ],
    "国籍": [
      {
        "subject": "查尔斯·阿兰基斯",
        "object": "智利"
      }
    ],
    "出生日期": [
      {
        "subject": "查尔斯·阿兰基斯",
        "object": "1989年4月17日"
      }
    ]
  }
]
```

模型训练和推理的详细代码见 [litie](https://github.com/xusenlinzy/lit-ie)