code: def get_models(num_clusters):
      """Get the two models we will be using."""
      # Hint: play with the number of layers to achieve different level of
      # over-fitting and observe its effects on membership inference performance.
      three_layer_model = tf.keras.Sequential(
          [
              tf.keras.layers.Dense(300, activation="relu"),
              tf.keras.layers.Dense(300, activation="relu"),
              tf.keras.layers.Dense(300, activation="relu"),
              tf.keras.layers.Dense(num_clusters, activation="relu"),
              tf.keras.layers.Softmax(),
          ]
      )
      three_layer_model.compile(
          optimizer="adam",
          loss=tf.keras.losses.SparseCategoricalCrossentropy(),
          metrics=["accuracy"],
      )
  
      two_layer_model = tf.keras.Sequential(
          [
              tf.keras.layers.Dense(300, activation="relu"),
              tf.keras.layers.Dense(300, activation="relu"),
              tf.keras.layers.Dense(num_clusters, activation="relu"),
              tf.keras.layers.Softmax(),
          ]
      )
      two_layer_model.compile(
          optimizer="adam",
          loss=tf.keras.losses.SparseCategoricalCrossentropy(),
          metrics=["accuracy"],
      )
      return three_layer_model, two_layer_model
  
file_path: privacy/tensorflow_privacy/privacy/privacy_tests/membership_inference_attack/codelabs/example.py
imports: import os
  import tempfile
  from absl import app
  from absl import flags
  import matplotlib.pyplot
  import numpy
  import pandas
  import scipy.stats
  from sklearn import metrics
  import tensorflow
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack import data_structures
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack import membership_inference_attack
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack import plotting
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack import privacy_report
label_desc: Use image node_id 3 for calling a global function with example usage: get_models(num_clusters) and returns: three_layer_model,  two_layer_model