search

harryjo97 / GruM

Official Code Repository for the paper "Graph Generation with Diffusion Mixture" (ICML 2024).
https://arxiv.org/abs/2302.03596
22 stars 2 forks source link

Adapt codebase to new 3D graph dataset #3

Open kdmayer opened 1 month ago

kdmayer commented 1 month ago

Hi,

Thank you for the great work!

Just wondering whether you would be willing to share any insights into how to best adapt the codebase to a new 3D graph dataset. We have a dataset of 3D building wireframes, where each building wireframe is represented as a list of 3D points and the corresponding adjacency matrix.

Would you recommend building on top of GruM-2D or GruM-3D? What would be your suggested approach?

Intuitively, I would think GruM-3D, but GruM-3D seems heavily geared towards molecule datasets and your README states that GruM-2D is better suited for general graph modeling tasks.

Thank you, Kevin

kdmayer commented 3 weeks ago

Would appreciate any thoughts or suggestions you have on this.