code: def __init__(
      self, metadata, hidden_channels, num_layers, output_channels
  ):
      super().__init__()
      self.model = to_hetero(
          GraphSAGE(
              (-1, -1), hidden_channels, num_layers, output_channels
          ),
          metadata,
      )
  
file_path: pytorch_geometric/benchmark/utils/hetero_sage.py
imports: import torch
  from tqdm import tqdm
  from torch_geometric.nn import GraphSAGE, to_hetero
label_desc: Use image node_id 1 to create a new HeteroGraphSAGE object from inherited base classes: torch.nn with example: obj = HeteroGraphSAGE(metadata, hidden_channels, num_layers, output_channels)