code: def __init__(
      self,
      node_type,
      edge_type,
      max_id,
      dim,
      num_negs=5,
      metric="mrr",
      **kwargs,
  ):
      self.node_type = node_type
      self.edge_type = edge_type
      self.max_id = max_id
      self.num_negs = num_negs
      self.dim = dim
      self.kernel = tf_euler.utils.layers.Dense(dim, use_bias=False)
      self.metric_class = tf_euler.utils.metrics.get(metric)
      self.metric_name = metric
  
file_path: euler/examples/dgi/dgi.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  import tf_euler
label_desc: Use image node_id 1 to create a new DGIBase object from inherited base classes: object with example: obj = DGIBase(node_type, edge_type, max_id, dim, num_negs, metric)