class_docstr: load the FB15K
docstr: load kg dataset.
code: def load_data(self):
      """
      load kg dataset.
      """
      log.info("Start loading the {} dataset".format(self.name))
      with open(
          os.path.join(self._data_dir, "entity2id.txt"), "r"
      ) as f:
          for line in f.readlines():
              line = line.strip().split("\t")
              self._entity2id[line[0]] = int(line[1])
      with open(
          os.path.join(self._data_dir, "relation2id.txt"), "r"
      ) as f:
          for line in f.readlines():
              line = line.strip().split("\t")
              self._relation2id[line[0]] = int(line[1])
      self._triple_train = self.load_kg_triple("train.txt")
      self._triple_test = self.load_kg_triple("test.txt")
      self._triple_valid = self.load_kg_triple("valid.txt")
  
      self.relation_total = len(self._relation2id)
      self.entity_total = len(self._entity2id)
      self.train_num = len(self._triple_train)
      self.test_num = len(self._triple_test)
      self.valid_num = len(self._triple_valid)
  
      # bloom_capacity = len(self._triple_train) + len(self._triple_test) + len(self._triple_valid)
      # self.training_triple_pool = BloomFilter(capacity=bloom_capacity, error_rate=0.01)
      for i in range(len(self._triple_train)):
          self.training_triple_pool.add(
              (
                  self._triple_train[i, 0],
                  self._triple_train[i, 1],
                  self._triple_train[i, 2],
              )
          )
  
      for i in range(len(self._triple_test)):
          self.training_triple_pool.add(
              (
                  self._triple_test[i, 0],
                  self._triple_test[i, 1],
                  self._triple_test[i, 2],
              )
          )
  
      for i in range(len(self._triple_valid)):
          self.training_triple_pool.add(
              (
                  self._triple_valid[i, 0],
                  self._triple_valid[i, 1],
                  self._triple_valid[i, 2],
              )
          )
      log.info("entity number: {}".format(self.entity_total))
      log.info("relation number: {}".format(self.relation_total))
      log.info("training triple number: {}".format(self.train_num))
      log.info("testing triple number: {}".format(self.test_num))
      log.info("valid triple number: {}".format(self.valid_num))
  
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 7 for calling the KGLoader obj's underlying member method code with example usage: obj.load_data() without return types