add data fetching model of link_prediction