search

aehrc / LAAT

A Label Attention Model for ICD Coding from Clinical Text
Other
66 stars 22 forks source link

Visualize the output of attention? #8

Closed Abhinav43 closed 2 years ago

Abhinav43 commented 2 years ago

Is it possible to explain the output of attention? Which token is contributing how much to the prediction? the depth of colour on each text represents the level of contribution such as bert visualizer etc?

tienthanhdhcn commented 2 years ago

@Abhinav43 this might be the code you want predictor_with_attention.zip

Abhinav43 commented 2 years ago

@tienthanhdhcn Thank you very much for the quick response. Could you please explain how to use it ? Any demo or use script? Which function to run first? Also, while going through the code, I couldn't find "code_desc.txt" anywhere in the folder?