Not able to convert yhat or yhat_raw outputs into icd codes

[cascademars]

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.