NSAPH-Projects / topological-equivariant-networks

E(n)-Equivariant Topological Neural Networks
MIT License
19 stars 0 forks source link

Add neighbor types #29

Closed ekarais closed 6 months ago

ekarais commented 6 months ago

This PR addresses #14 by introducing a new script argument neighbor_type: str that allows defining when two cells of same rank will be considered neighbors. The possible values for this argument are:

If we add a supernode representing the graph and pick either any_adjacency or all, all cells with the same rank will be connected to each other.

After writing unit tests for the new methods, our code coverage increased to 37% :partying_face: