code: def __init__(
      self,
      dims,
      metapath,
      label_dim,
      feature_idx,
      feature_max_id,
      processing_steps=4,
      lstm_layers=2,
  ):
      super(Set2SetModel, self).__init__(label_dim)
      self.gnn = GNN(
          "sage",
          "full",
          dims,
          None,
          metapath,
          feature_idx,
          feature_max_id,
      )
      self.pool = tf_euler.graph_pool.Set2SetPool(
          dims[-1],
          processing_steps=processing_steps,
          num_layers=lstm_layers,
      )
  
file_path: euler/examples/set2set/set2set.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  import tf_euler
  from tf_euler.python.mp_utils.base_gnn import BaseGNNNet
  from tf_euler.python.mp_utils.base_graph import GraphModel
label_desc: Use image node_id 1 to create a new Set2SetModel object from inherited base classes: GraphModel with example: obj = Set2SetModel(dims, metapath, label_dim, feature_idx, feature_max_id, processing_steps, lstm_layers)