search

LuChang-CS / sherbet

Self-supervised graph learning with hyperbolic embedding for temporal health event prediction (IEEE Cybernetics)
https://ieeexplore.ieee.org/document/9543467
Apache License 2.0
13 stars 0 forks source link

Potential sharing of the ICD-10 embedding? #1

Open LechengKong opened 1 year ago

LechengKong commented 1 year ago

Thanks for sharing the source code, and I think this is an interesting work to embed ICD codes. Would you mind sharing the ICD-10 embedding if it is not confidential? Thanks!

LuChang-CS commented 1 year ago

Hi, I appreciate your interest in this work. But in this project, we use MIMIC-III and eICU with ICD-9 instead of ICD-10.

LechengKong commented 1 year ago

Thanks for the reply! Then, do you still keep the ICD9 weight? If so, do you mind sharing?

LuChang-CS commented 1 year ago

I have uploaded the embeddings and a disease -> code map: Google Drive. You can use pickle.load to access them. Note: this file is not trained with the full hierarchical structure. We do some compression in the structure.

LechengKong commented 1 year ago

Thank you for your help in such a short time! You are a great Open source developer!