Imported from https://github.com/matejklemen/got-link-prediction