code: def __init__(self, data_dir=fb_dir, data_type="all"):
      super(FB15K237, self).__init__(data_dir, data_type)
      self.source_url = "https://download.microsoft.com/download/8/7/0/8700516A-AB3D-4850-B4BB-805C515AECE1/FB15K-237.2.zip"
      self.origin_files = [
          "Release/test.txt",
          "Release/train.txt",
          "Release/valid.txt",
      ]
      self.max_node_id = 14541
      self.max_edge_id = 237
      self.train_node_type = ["train"]
      self.train_edge_type = ["train"]
      self.total_size = 310116
      self.all_node_type = -1
      self.all_edge_type = ["train", "valid", "test"]
      self.edge_id_file = os.path.join(
          data_dir, "fb15k-237_test.edgeid"
      )
      self.node_id_file = os.path.join(
          data_dir, "fb15k-237_test.nodeid"
      )
      self.edge_id_idx = "id"
      self.edge_id_dim = 1
  
file_path: euler/tf_euler/python/dataset/fb15k237.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import json
  import urllib
  import zipfile
  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 FB15K237 object from inherited base classes: DataSet with example: obj = FB15K237(data_dir, data_type)