code: def numerical_test():
      x_train, y_train, x_test, y_test = build_mnist_data(NUM_CLASSES)
      keras_model = build_keras_model(keras, NUM_CLASSES)
      tf_keras_model = build_keras_model(tf_keras, NUM_CLASSES)
  
      # Make sure both model have same weights before training
      weights = [weight.numpy() for weight in keras_model.weights]
      tf_keras_model.set_weights(weights)
  
      for kw, kcw in zip(keras_model.weights, tf_keras_model.weights):
          np.testing.assert_allclose(kw.numpy(), kcw.numpy())
  
      keras_history = train_model(keras_model, x_train, y_train)
      tf_keras_history = train_model(tf_keras_model, x_train, y_train)
  
      for key in keras_history.history.keys():
          np.testing.assert_allclose(
              keras_history.history[key],
              tf_keras_history.history[key],
              atol=1e-3,
          )
  
file_path: keras/integration_tests/numerical_test.py
imports: import keras
  import numpy
  from tensorflow import keras
label_desc: Use image node_id 5 for calling a global function with example usage: numerical_test() without return types