[Feature enhancement] Edge handling in GCNEnc
Junyoungpark opened this issue · comments
Junyoung Park commented
I found that the current implementation of GCNEnc cannot support the cases where we test the model that is not trained.
This could be critical especially when we measure the generalization performance of the trained model.
An easy fix is, inside of the forward, when the given x.shape[1] != self.edge_index[0].max() - or something more smart -, we can reconstruct the edges from the inputs.
Chuanbo HUA commented
Good idea, added this feature in 1c22b6e.