how to get the diffusion map of paper

[ziwenjie]

Hi:
The diffusion map is so pretty, but I don't know how to get it.

May I have your help.
Best regard.

[chirag-sj]

You can refer to this repo: https://github.com/Shanthika/heat_diffusion/blob/main/code.ipynb
Direct Implicit Timestep is used for visualizing the kernel.