Predict Head or Tail candidates using GraphSAGE model

[geo47]

Hi

How to train GraphSAGE model for prediction tasks, like TransE, RotateE, ComplEx models etc.

For example: After training GraphSAGE on KG predict head or tail entities using query_topn

Thanks