Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add LineDigraph transformation #9592

Open
wants to merge 12 commits into
base: master
Choose a base branch
from

Conversation

petern48
Copy link

Addresses #9457 in collaboration with @Flunzmas

Changes:

  • Implement an optimized transforms module for LineDiGraph based off of the implementation by Bernárdez et al.
  • Add a test for testing the implementation

Motivation from Flunzmas:

Recent scientific work [1] leveraging GNNs for routing in computer networks has used a line digraph representation of the computer network topology as input for their GNN policies. The line digraph is formed by taking the original graph's edges as new nodes, and drawing an edge between those new nodes that, as edges in the original graph, form a directed path of length two: E' = {(u, v), (w, x) | (u, v) in E; (w, x) in E; v = w}.

[1]: Bernárdez et al. "Magnneto: A graph neural network-based multi-agent system for traffic engineering." IEEE Transactions on Cognitive Communications and Networking 9.2 (2023)

@petern48 petern48 requested a review from wsad1 as a code owner August 14, 2024 14:10
torch_geometric/transforms/line_digraph.py Outdated Show resolved Hide resolved
test/transforms/test_line_digraph.py Outdated Show resolved Hide resolved
torch_geometric/transforms/line_digraph.py Outdated Show resolved Hide resolved
@petern48 petern48 requested a review from Kh4L August 26, 2024 14:40
@petern48
Copy link
Author

petern48 commented Sep 9, 2024

@rusty1s Do I need to do anything to update the docs? I thought it would produce the page automatically, but I don't see it in the built docs.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants