harryjo97 / GDSS

Official Code Repository for the paper "Score-based Generative Modeling of Graphs via the System of Stochastic Differential Equations" (ICML 2022)
https://arxiv.org/abs/2202.02514
142 stars 22 forks source link

Question on the data format for generated samples of ENZYMES dataset #14

Open hash-wls opened 1 year ago

hash-wls commented 1 year ago

Thank you for your excellent work, I have a small question. I want to use generated samples from ENZYMES dataset for graph classification task, where each graph should have a corresponding label, can your work generate graph labels?

harryjo97 commented 1 year ago

Our work generates node features and adjacency together. In order to generate graph labels, you can add the graph labels to the node features (i.e., adding the graph label to the feature of nodes along with the one-hot encoded degrees).