Coding the Solution

The solution to the problem

Our aim is to generate friend recommendations for different people in the social network graph. We can formulate this as a link prediction or a knowledge graph completion task. Specifically, this is a task of head prediction or tail prediction.

Let's have a look at the following code and run it to generate the results: