code: def mymodel(shape):
      x_in = Input(shape=shape)
      x = Dense(n_hidden)(x_in)
      x_out = Dense(n_classes, activation="softmax")(x)
      return tf.keras.models.Model(inputs=x_in, outputs=x_out)
  
file_path: alibi-detect/alibi_detect/cd/tests/test_ks.py
imports: from functools import partial
  from itertools import product
  import numpy
  import pytest
  import tensorflow
  from tensorflow.keras.layers import Dense, Input, InputLayer
  from typing import Callable
  from alibi_detect.cd import KSDrift
  from alibi_detect.cd.tensorflow.preprocess import HiddenOutput, UAE, preprocess_drift
label_desc: Use image node_id 1 for calling a global function with example usage: mymodel(shape) and returns: tf