alan-turing-institute / grace

Graph Representation Analysis for Connected Embeddings
Other
34 stars 0 forks source link

[DEVELOPMENT] Consider using a KNN to construct the intial graph #337

Open quantumjot opened 10 months ago

quantumjot commented 10 months ago

We should consider using a KNN to construct the intial graph, and compare this with the current triangulation method.

https://pytorch-geometric.readthedocs.io/en/latest/generated/torch_geometric.nn.pool.knn_graph.html#torch_geometric.nn.pool.knn_graph

This would have the additional benefit of allowing alternative hypotheses (such as crossing) for object connections in the optimisation step, at the cost of additional computational complexity.