similarity calculation

[BudBudding]

非常感谢您的工作！关于您的工作中similarity calculation这个图的生成

请问是怎么获得的哇

[Spico197]

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

论文中的图是一个示意图，可以根据需要编一个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一下啥的），打印中间值：

DocEE/dee/models/trigger_aware.py

Line 595 in a32b6f8

torch.sigmoid(scores).ge(self.config.biaffine_hard_threshold).long()