Code for "Distributed, Egocentric Representations of Graphs for Detecting Critical Structures" (ICML 2019)