main.py: the core of our model, including the structure and the process of training. calculator.py: the code about calculating Poincare embedding and class-aware Ricci curvature weights dataloader.py: ...