README.md

---
language:
- zh
tags:
- event extraction
license: apache-2.0
datasets:
- DuEE
metrics:
- f1
---

# GPLinker事件抽取模型

## 模型介绍

+ 数据集：百度 `DUEE` 信息抽取

+ 模型方法：[GPLinker：基于GlobalPointer的事件联合抽取](https://spaces.ac.cn/archives/8926)

## 使用方法

```commandline
pip install litie
```

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

pipeline = EventExtractionPipeline("gplinker", model_name_or_path="xusenlin/duee-gplinker", model_type="bert")
text = "油服巨头哈里伯顿裁员650人 因美国油气开采活动放缓。"
pprint(pipeline(text))

# 输出
[
  [
    {
      "event_type": "组织关系-裁员",
      "arguments": [
        {
          "role": "裁员人数",
          "argument": "650人"
        },
        {
          "role": "裁员方",
          "argument": "油服巨头哈里伯顿"
        }
      ]
    }
  ]
]
```

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