Poincare embedding for ICD coding pytorch

• 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 and RiemannianSGD 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.