class_docstr: Load WikiKG90M from files.
      
class_docstr_tok: ["Load","WikiKG90M","from","files","."]
code: def __init__(self, path):
      super(WikiKG90MDataset, self).__init__()
      self.name = "WikiKG90M-LSC"
      try:
          from ogb.lsc import WikiKG90MDataset as LSCDataset
      except ImportError as error:
          print(
              "Please run ``pip install ogb==1.3.1`` to load WikiKG90M dataset."
          )
          raise ImportError(error)
      data = LSCDataset(path)
      valid = data.valid_dict["h,r->t"]
      test = data.test_dict["h,r->t"]
      self.triplets = {
          "train": data.train_hrt,
          "valid": {
              "mode": "wikikg90m",
              "h": valid["hr"][:, 0],
              "r": valid["hr"][:, 1],
              "candidate_t": valid["t_candidate"],
              "t_correct_index": valid.get("t_correct_index", None),
          },
          "test": {
              "mode": "wikikg90m",
              "h": test["hr"][:, 0],
              "r": test["hr"][:, 1],
              "candidate_t": test["t_candidate"],
              "t_correct_index": test.get("t_correct_index", None),
          },
      }
      self.num_ents = data.num_entities
      self.num_rels = data.num_relations
      self.ent_feat = data.entity_feat
      self.rel_feat = data.relation_feat
  
imports: import os
  import time
  import numpy
  import paddle.distributed
  from dataset.trigraph import TriGraph
  from utils import timer_wrapper
label_desc: Use image node_id 1 to create a new WikiKG90MDataset object from inherited base classes: object with example: obj = WikiKG90MDataset(path)