Update pages/15_Graphs.py

pages/15_Graphs.py

@@ -13,7 +13,7 @@ os.environ['TF_USE_LEGACY_KERAS'] = '1'
 def model_fn(graph_tensor_spec: tfgnn.GraphTensorSpec):
     graph = inputs = tf.keras.Input(type_spec=graph_tensor_spec)
 
-    # Encode input features (callback omitted for brevity).
+    # Encode input features
     graph = tfgnn.keras.layers.MapFeatures(
         node_sets_fn=lambda node_set, node_set_name: tf.keras.layers.Dense(64)(node_set['features'])
     )(graph)
@@ -26,7 +26,7 @@ def model_fn(graph_tensor_spec: tfgnn.GraphTensorSpec):
             attention_type="none", simple_conv_reduce_type="mean",
             normalization_type="layer", next_state_type="residual",
             state_dropout_rate=0.2, l2_regularization=1e-5,
-            receiver_tag=tfgnn.NODES  # Correctly use tfgnn.NODES
+            receiver_tag=tfgnn.TARGET  # Use TARGET instead of NODES
         )(graph)
 
     return tf.keras.Model(inputs, graph)