How to reproduce the visualization part of the paper?

HazyResearch / hgcn

Hyperbolic Graph Convolutional Networks in PyTorch.

602 stars 109 forks source link

How to reproduce the visualization part of the paper? #2

Closed feverer closed 4 years ago

feverer commented 4 years ago

Hi, I read your paper and I esperciall like the visualization of embedding in your job, I expected to reproduce the result on cora, where different classes in cora can be separated well. Could you please show me how to achieve that?

nghiahhnguyen commented 4 years ago

I think that such visualization can be done with t-SNE because that's how other papers do it.

ines-chami commented 4 years ago

Hi, for visualizations we used the Poincare Ball model with dimension set to 2, and then colored node embeddings based on the corresponding label class.