Graph embeddings and graph classification

marius-team / marius

Large scale graph learning on a single machine.

https://marius-project.org

Apache License 2.0

160 stars 45 forks source link

Graph embeddings and graph classification #98

Closed bkrdmr closed 2 years ago

bkrdmr commented 2 years ago

Hi, I've been reading the documentation for node classification and edge prediction tasks. I have a set of custom graphs I'd like to use for graph classification or graph-level embeddings for additional downstream tasks. Is this possible with the current version of Marius? Thank you.

JasonMoho commented 2 years ago

Unfortunately we don't currently support the graph classification task. That's something we have planned for the longer term.

Out of curiosity, how many graphs are in the dataset and what is their rough size (nodes/edges)? I don't have much familiarity with the graph classification workload beyond the OGB datasets.