/Stellar-Graph-Link-Prediction

Link prediction on a created graph using stellar graph library with different methods: GraphSAGE, HinSAGE, Metapath2Vec. TODO: Codes need to be organized.

Primary LanguagePython

Stellar-Graph-Link-Prediction

Link prediction on a created graph using stellar graph library with different methods: GraphSAGE, HinSAGE, Metapath2Vec.

TODO: Codes need to be organized. Data portions should be generalized.

Sets are arranged by hand, can be used if train/test portions are already seperated and it is important to use the same partitions.