- pytorch implementation of poincare embedding to ICD coding hierarchy(chinese version of ICD-9-CM3 procedure codes hierarchy), similar to Poincaré embeddings for learning hierarchical representations, the idea of modelling ICD hierarchy in hypobolic space is taken from HyperCore: Hyperbolic and Co-graph Representation for Automatic ICD Coding.
- Dependencies:
- pytorch
- treelib
- tqdm
- matplotlib
manifolds
package andRiemannianSGD
function are taken from facebook's implementation:- run
ICD_main.py
to get 2d visualization of ICD hierarchy on the Poincare ball. - This code is licensed under CC-BY-NC 4.0.