https://medium.com/@monocosmo77/applications-of-graph-neural-networks-part3-machine-learning-57b71405d302