def METHOD_NAME(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))