tomdbar/ecord

Training configuration of ecord

Closed this issue · 0 comments

Hello,
I want to train an ecord model that replicates the results in your paper on GSET. I think that model was trained on ER500 instances, so if I am not mistaken I should change
graph_generator=ErdosRenyiGraphGenerator([40,40], (0.15, 0.0), EdgeType.SIGNED) to
graph_generator=ErdosRenyiGraphGenerator([500,500], (0.15, 0.0), EdgeType.SIGNED).
And also change training config as described in your paper at part "Training details".