How to handle multiple graph lables

,

Hello,

My graph comes with multiple labels and it’s not easy to organize them into a tensor. For example g.a is a scalar, g.b is a vector and g.c is a matrix…

I can save each individual property as a key and access them by key. But When I try to save and load them following https://docs.dgl.ai/guide/data-savenload.html#guide-data-pipeline-savenload, I found the key I created is lost. It seems that save_graph will not save the extra key-value user created.

So I am looking for a more elegant way to handle the graph of multiple labels. The link above demonstrates only graph with a single label (a scalar) but I have a graph with multiple labels (best organized as key-value pair in a dict). Do you have some recommendations for my case? Thanks!

Hi,

If it’s not tensor, you should use save_info instead. It’s just pickle and will store everything

Thank you very much!