A vertex prediction model implementation based on https://arxiv.org/abs/1704.05548