cynricfu / MAGNN

Metapath Aggregated Graph Neural Network for Heterogeneous Graph Embedding
395 stars 69 forks source link

How can I modify the code so that only MAGNN is used to get the node embedding representation #37

Closed Yuruzu closed 1 year ago

Yuruzu commented 1 year ago

Thanks for sharing the code, I would like to use the MAGNN graph embedding algorithm to get the embedded representation of the nodes, and then do the follow-up work with other models. When I use base_MAGNN.py, do I also need to process the data into the form of your preprocessing? Could you please give me some ideas? Thank you

cynricfu commented 1 year ago

You would need to preprocess your data in a way similar to our preprocessing code. Then you can apply MAGNN to train upon your data, using MAGNN_nc for node classification or MAGNN_lp for link prediction. Finally, you may use the trained model to obtain the node representations.