search

GanjinZero / ICD-MSMN

Code Synonyms Do Matter: Multiple Synonyms Matching Network for Automatic ICD Coding [ACL 2022]
https://arxiv.org/abs/2203.01515
50 stars 8 forks source link

Not able to convert yhat or yhat_raw outputs into icd codes #10

Open cascademars opened 1 year ago

cascademars commented 1 year ago

Hi. Thanks for the great work. Im trying to write a inference script for this repo. But not able to understand how to convert yhat or yhat_raw predictions into ICD codes. If possible can you please suggest how to go ahead. Thanks.

GanjinZero commented 1 year ago

Please see load_full_codes(self.train_path, version=version) in data_util.py to convert label id to icd codes.