code: def __init__(self, multi_out: bool = False):
      super(MyModel, self).__init__()
      self.dense = Dense(n_classes, activation="softmax")
      self.multi_out = multi_out
  
file_path: alibi-detect/alibi_detect/utils/tensorflow/tests/test_prediction_tf.py
imports: import numpy
  import pytest
  import tensorflow
  from tensorflow.keras.layers import Dense, InputLayer
  from typing import Tuple, Union
  from alibi_detect.utils.tensorflow import predict_batch
label_desc: Use image node_id 1 to create a new MyModel object from inherited base classes: tf.keras with example: obj = MyModel(multi_out)