class_docstr: Implementation of GATNE model.
  
  Args:
      config: dict, some configure parameters.
      dataset: instance of Dataset class
      place: GPU or CPU place 
class_docstr_tok: ["Implementation","of","GATNE","model",".","Args",":","config",":","dict",",","some","configure","parameters",".","dataset",":","instance","of","Dataset","class","place",":","GPU","or","CPU","place"]
code: def __init__(self, config, dataset, place):
      logging.info(["model is: ", self.__class__.__name__])
      self.config = config
      self.graph = dataset.graph
      self.placce = place
      self.edge_types = sorted(self.graph.edge_types_info())
      logging.info("edge_types in model: %s" % str(self.edge_types))
      neg_num = dataset.config["neg_num"]
  
      # hyper parameters
      self.num_nodes = self.graph.num_nodes
      self.embedding_size = self.config["dimensions"]
      self.embedding_u_size = self.config["edge_dim"]
      self.dim_a = self.config["att_dim"]
      self.att_head = self.config["att_head"]
      self.edge_type_count = len(self.edge_types)
      self.u_num = self.edge_type_count
  
      self.gw = heter_graph_wrapper.HeterGraphWrapper(
          name="heter_graph",
          edge_types=self.graph.edge_types_info(),
          node_feat=self.graph.node_feat_info(),
          edge_feat=self.graph.edge_feat_info(),
      )
  
      self.train_inputs = fl.data(
          "train_inputs", shape=[None], dtype="int64"
      )
  
      self.train_labels = fl.data(
          "train_labels", shape=[None, 1, 1], dtype="int64"
      )
  
      self.train_types = fl.data(
          "train_types", shape=[None, 1], dtype="int64"
      )
  
      self.train_negs = fl.data(
          "train_negs", shape=[None, neg_num, 1], dtype="int64"
      )
  
      self.forward()
  
file_path: PGL/legacy/examples/GATNE/model.py
imports: import numpy
  import math
  import logging
  import paddle.fluid
  import paddle.fluid.layers
  from pgl import heter_graph_wrapper
label_desc: Use image node_id 1 to create a new GATNE object from inherited base classes: object with example: obj = GATNE(config, dataset, place)