List of nodes

[mardukevich]

Previosly i asked you on https://medium.com/octavian-ai/finding-shortest-paths-with-graph-networks-807c5bbfc9c8 , but you didn't answer. Is this neural network can return the path ? I mean the list of nodes, not just a count of nodes. Maybe i didn’t understand something.

[TianrenWang]

This specific example cannot return you the list of nodes because the answer the network was trained on was the count of nodes. However I believe it is possible to return the actual path of the shortest path by replacing all of the answers with the result from running every graph in the dataset with Dijkstra's algorithm.

EDIT: TLDR

1. Run Dijkstra's algorithm on every graph used for training and get the list of nodes
2. Replace the answer to every question with the list of nodes you got in step 1.