use node index instead of ID in alignment matrix

snap-stanford / neural-subgraph-learning-GNN

329 stars 64 forks source link

use node index instead of ID in alignment matrix #13

Closed cgoliver closed 3 years ago

cgoliver commented 3 years ago

Hi! gen_alignment_matrix() crashes with an index error when the node IDs of the input graphs don't correspond to indices in the A matrix. Added an enumerate in the for loops to prevent this.

qema commented 3 years ago

Looks great, thanks for the catch and fix!