Repo in paper - https://github.com/stair-lab/kg-gen/