code: def main(
      model_name,
      dataset,
      entity_feat,
      relation_feat,
      entity_neighbors,
      alpha=0.98,
      k_hop=10,
      gamma=6.0,
      degree_w=0.1,
      r_emb=None,
      r_emb_mat=None,
      ote_size=20,
      indegrees=None,
      neighbor_norm=False,
      scale_norm=False,
  ):
      for i in range(k_hop):
          start = time.time()
          if model_name == "TransE":
              entity_feat = rep_transe(
                  entity_feat,
                  relation_feat,
                  entity_neighbors,
                  alpha=alpha,
                  degree_w=degree_w,
                  indegrees=indegrees,
                  neighbor_norm=neighbor_norm,
              )
          elif model_name == "RotatE":
              entity_feat = rep_rotate(
                  entity_feat,
                  relation_feat,
                  entity_neighbors,
                  alpha=alpha,
                  degree_w=degree_w,
                  gamma=gamma,
                  indegrees=indegrees,
                  neighbor_norm=neighbor_norm,
              )
          elif model_name == "OTE":
              entity_feat = rep_ote(
                  entity_feat,
                  relation_feat,
                  entity_neighbors,
                  alpha=alpha,
                  degree_w=degree_w,
                  r_emb=r_emb,
                  r_emb_mat=r_emb_mat,
                  ote_size=ote_size,
                  indegrees=indegrees,
                  neighbor_norm=neighbor_norm,
                  scale_norm=scale_norm,
              )
          elif model_name == "DistMult":
              entity_feat = rep_distmult(
                  entity_feat,
                  relation_feat,
                  entity_neighbors,
                  alpha=alpha,
                  degree_w=degree_w,
                  indegrees=indegrees,
                  neighbor_norm=neighbor_norm,
              )
          end = time.time()
          print(
              "Time elapsed for running one hop: %.4f" % (end - start)
          )
      save_path = "REP_save_feat_%s_%s" % (model_name, dataset)
      if not os.path.exists(save_path):
          os.mkdir(save_path)
          np.save(
              os.path.join(save_path, "entity_embedding.npy"),
              entity_feat,
          )
          np.save(
              os.path.join(save_path, "relation_embedding.npy"),
              relation_feat,
          )
  
file_path: PGL/apps/Graph4KG/examples/REP/rep.py
imports: import os
  import time
  import logging
  import argparse
  import numpy
  from tqdm import tqdm
  import pgl
  from ote_orth import OrthOTE
label_desc: Use image node_id 8 for calling a global function with example usage: main(model_name, dataset, entity_feat, relation_feat, entity_neighbors, alpha, k_hop, gamma, degree_w, r_emb, r_emb_mat, ote_size, indegrees, neighbor_norm, scale_norm) without return types