code: def __init__(self, data_dir=fb_dir, data_type="all"):
      super(FB15K, self).__init__(data_dir, data_type)
      self.source_url = "https://everest.hds.utc.fr/lib/exe/fetch.php?media=en:fb15k.tgz"
      self.origin_files = [
          "freebase_mtr100_mte100-test.txt",
          "freebase_mtr100_mte100-train.txt",
          "freebase_mtr100_mte100-valid.txt",
      ]
      self.max_node_id = 14951
      self.max_edge_id = 1345
      self.train_node_type = ["train"]
      self.train_edge_type = ["train"]
      self.total_size = 592213
      self.all_node_type = -1
      self.all_edge_type = ["train", "valid", "test"]
      self.edge_id_file = os.path.join(data_dir, "fb15k_test.edgeid")
      self.node_id_file = os.path.join(data_dir, "fb15k_test.nodeid")
      self.edge_id_idx = "id"
      self.edge_id_dim = 1
  
file_path: euler/tf_euler/python/dataset/fb15k.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import json
  import urllib
  import tarfile
  import tqdm
  from tf_euler.python.dataset.sage_util import *
  from tf_euler.python.dataset.base_dataset import DataSet
label_desc: Use image node_id 1 to create a new FB15K object from inherited base classes: DataSet with example: obj = FB15K(data_dir, data_type)