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
146 stars 22 forks source link

Simple Conditional Generation #25

Open xwang112358 opened 7 months ago

xwang112358 commented 7 months ago

Dear authors,

Thank you for the great work. In my case, I want to do a simple conditional generation task, where each graph is associated with a label from {0,1}. During the sampling process, I would like to control the generated graph with the label. My initial though is to add a new parameter for label in two score models(GNNs) during training. Could you tell me how you will realize this task?

Best, Allen Wnag