code: def mymodel(shape, softmax: bool = True):
      x_in = Input(shape=shape)
      x = Dense(20, activation=tf.nn.relu)(x_in)
      x = Dense(2)(x)
      if softmax:
          x = tf.nn.softmax(x)
      return tf.keras.models.Model(inputs=x_in, outputs=x)
  
file_path: alibi-detect/alibi_detect/cd/tensorflow/tests/test_classifier_tf.py
imports: from itertools import product
  import numpy
  import pytest
  import tensorflow
  from tensorflow.keras.layers import Dense, Input
  from typing import Union
  from alibi_detect.cd.tensorflow.classifier import ClassifierDriftTF
label_desc: Use image node_id 1 for calling a global function with example usage: mymodel(shape, softmax) and returns: tf