code: def build_graph():
      keras.backend.clear_session()
      image_input = ak.ImageInput(shape=(32, 32, 3))
      image_input.batch_size = 32
      image_input.num_samples = 1000
      merged_outputs = ak.SpatialReduction()(image_input)
      head = ak.ClassificationHead(num_classes=10, shape=(10,))
      classification_outputs = head(merged_outputs)
      return ak.graph.Graph(
          inputs=image_input, outputs=classification_outputs
      )
  
file_path: autokeras/autokeras/test_utils.py
imports: import inspect
  import os
  import numpy
  import tensorflow
  from tensorflow import keras
  import autokeras
label_desc: Use image node_id 5 for calling a global function with example usage: build_graph() and returns: ak