code: def _build_toy_model(self):
      # 1-input, 2-output linear model (logit = w * x)
      return tf.keras.Sequential(
          [
              tf.keras.Input(shape=[1], dtype=tf.float32),
              tf.keras.layers.Dense(2, use_bias=False),
          ]
      )
  
file_path: neural-structured-learning/research/multi_representation_adversary/multi_representation_adversary/evaluator_test.py
imports: import os
  from absl.testing import parameterized
  import gin
  from multi_representation_adversary import evaluator
  from multi_representation_adversary import resnet
  import numpy
  import tensorflow.compat.v2
  import tensorflow_datasets
label_desc: Use image node_id 2 for calling the AggregationTest obj's underlying member method code with example usage: obj._build_toy_model() and returns: tf