class_docstr: load the FB15K
docstr: init
code: def __init__(self, data_dir, batch_size, neg_mode, neg_times):
      """init"""
      self.name = os.path.split(data_dir)[-1]
      self._feed_list = ["pos_triple", "neg_triple"]
      self._data_dir = data_dir
      self._batch_size = batch_size
      self._neg_mode = neg_mode
      self._neg_times = neg_times
  
      self._entity2id = {}
      self._relation2id = {}
      self.training_triple_pool = set()
  
      self._triple_train = None
      self._triple_test = None
      self._triple_valid = None
  
      self.entity_total = 0
      self.relation_total = 0
      self.train_num = 0
      self.test_num = 0
      self.valid_num = 0
  
      self.load_data()
  
file_path: PGL/legacy/examples/pgl-ke/data_loader.py
imports: import os
  import numpy
  from collections import defaultdict
  from pgl.utils.logger import log
label_desc: Use image node_id 1 to create a new KGLoader object  with example: obj = KGLoader(data_dir, batch_size, neg_mode, neg_times)