code: def load_dataset(filename):
      """Reads a file in the `.tfrecord` format.
  
      Args:
        filename: Name of the file containing `tf.train.Example` objects.
  
      Returns:
        An instance of `tf.data.TFRecordDataset` containing the `tf.train.Example`
        objects.
      """
      return tf.data.TFRecordDataset([filename])
  
file_path: neural-structured-learning/neural_structured_learning/examples/graph_keras_mlp_cora.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl import app
  from absl import flags
  from absl import logging
  import attr
  import neural_structured_learning
  import tensorflow
label_desc: Use image node_id 2 for calling a global function with example usage: load_dataset(filename) and returns: tf