search

aidos-lab / pytorch-topological

A topological machine learning framework based on PyTorch
https://pytorch-topological.readthedocs.io/
BSD 3-Clause "New" or "Revised" License
142 stars 18 forks source link

Visualization of PD #35

Open sinAshish opened 5 months ago

sinAshish commented 5 months ago

Hi,

I feel like there should be in-built functions for visualizing Persistence diagrams.

Like now, I am using matplotlib, to scatter plot the diagram of PersistenceInformation, but it requires extra steps of detaching and blah blah. Plus, I dont know what would be the best way to visualize the pairing? So, I feel there should be something in the documentation or tutorial on how to do so!

Pseudomanifold commented 5 months ago

Hello!

That's a good idea and would make for a good PR. I'd suggest to solve this as an optional utility function, though, permitting users some way to customise the output. Let me know what you think!

sinAshish commented 5 months ago

I would be happy to make a PR for that. I am still in the learning phase of working with topological measures, so I am not quite sure what is the best way to visualize Persistence pairings. I can visualize the birth/death rates of PDs though.

Pseudomanifold commented 4 months ago

I think there's no agreed-upon visualisation of persistence pairings at this point. You could visualise them via a heat map or matrix, for instance.