Take Model as API to Extract event in Document

Spico197 / DocEE

🕹️ A toolkit for document-level event extraction, containing some SOTA model implementations.

https://doc-ee.readthedocs.io/

MIT License

234 stars 36 forks source link

Closed Ricardokevins closed 2 years ago

Ricardokevins commented 2 years ago

您好，我是做其他的NLP任务的，但是对抽取文档里的Event很感兴趣，发现了您的工作

通读了README之后看到了很详细的复现方法，但是想问一下是否有公开已经训练的模型，以及inference的API。可以比较方便的直接作为一个数据的预处理方法，在自己的数据上，获得文档中的事件，而不需要重新训练和阅读代码呢？

非常感谢您的建议

Spico197 commented 2 years ago

您好，感谢您对本项目的关注。

关于已训练的模型权重：我们提供 PTPCG 的模型权重供大家复现，您可以在 README 中的 To Reproduce Results in Paper / PTPCG 一节中找到下载链接；
关于推理接口：我们提供一个预测接口，输入纯文本，输出结构化事件。您可以参考这里的代码：dee/tasks/dee_task.py#L1112

Ricardokevins commented 2 years ago

供 PTPCG 的模型权重供大家复现，您可以在 R

您好，确认一下是天翼网盘里的248M的Task Dump吗

Spico197 commented 2 years ago

您好，是的。

Ricardokevins commented 2 years ago

您好，是的。

感谢您的回复。我去试试

Ricardokevins commented 2 years ago

您好，是的。

最后还有一个问题，这个模型是在中文数据集上训练的吗？我看了论文里的两个数据集好像是中文的？（之前没注意。。。）

Spico197 commented 2 years ago

是的，ChFinAnn 和 DuEE-fin 是中文数据集。

Ricardokevins commented 2 years ago

好吧，有点遗憾，不过感谢您的开源，README很详细，回复issue也很及时