search

YoumiMa / dreeam

source code for {D}ocument-level {R}elation {E}xtraction with {E}vidence-guided {A}ttention {M}echanism
MIT License
75 stars 16 forks source link

The code of Heatmaps #10

Closed edwarddiou closed 1 year ago

edwarddiou commented 1 year ago

Hi , I am interested in your research. May i ask about the code of heatmaps in the end of the paper ? I cant find it in the NCRF++.

Thanks a lot

YoumiMa commented 1 year ago

Hi @edwarddiou, thanks for your interest in this project, and sorry for the late reply. I have uploaded a jupyter notebook for heatmap visualization. Our visualization is based on another repository here. Please make sure to modify several hard-coded file paths in the notebook.

I hope this update can solve your problem, thanks!

edwarddiou commented 1 year ago

Thanks for the reply. 非常感謝 擷取 image

能否再請教一下，paper中的BERT版本和RoberTa 版本的DREEAM (teacher)與加上Inference-stage Fusion，是使用scripts/run_bert.sh 的參數並訓練30 epoch就可達到的嗎? 還是需要訓練更多epoch呢?

非常感謝您的答覆

YoumiMa commented 1 year ago

@edwarddiou 感謝您的回覆。 直接使用scripts/run_bert.sh與scripts/run_roberta.sh的參數應該就可以達到您截圖中第一行與第三行的性能。如果您想要再現inference-stage fusion的性能，請在模型訓練完成之後運行scripts/isf_bert.sh與scripts/isf_roberta.sh。

希望這能解決您的問題。謝謝！