Spico197 / DocEE

🕹️ A toolkit for document-level event extraction, containing some SOTA model implementations.
https://doc-ee.readthedocs.io/
MIT License
232 stars 36 forks source link

similarity calculation #81

Open BudBudding opened 8 months ago

BudBudding commented 8 months ago

非常感谢您的工作!关于您的工作中similarity calculation这个图的生成 图片 请问是怎么获得的哇

Spico197 commented 8 months ago

嗨您好,感谢您对本项目的关注~

论文中的图是一个示意图,可以根据需要编一个case,然后参考这里的代码进行绘制:https://matplotlib.org/stable/gallery/images_contours_and_fields/image_annotated_heatmap.html#sphx-glr-gallery-images-contours-and-fields-image-annotated-heatmap-py

如果是需要真实case的具体值,则需要hook这里的代码(比如print一下啥的),打印中间值:https://github.com/Spico197/DocEE/blob/a32b6f8bf0b2fe28b9942b0b53eff461abf561ba/dee/models/trigger_aware.py#L595