code: def get_tf_yhat(model):
      if not (
          str(type(model)).startswith(
              "<class 'tensorflow.python.keras.engine"
          )
          or str(type(model)).startswith("<class 'keras.engine")
      ):
          return None
  
      if model.output_shape[1] == 1:
          return yhat_tf_regression, "regression"
      elif model.output_shape[1] == 2:
          return yhat_tf_classification, "classification"
      else:
          warn(
              "Tensorflow: Output shape of the predict method should not be greater than 2"
          )
          return yhat_tf_classification, "classification"
  
file_path: DALEX/python/dalex/dalex/_explainer/yhat.py
imports: from warnings import warn
  import numpy
label_desc: Use image node_id 4 for calling a global function with example usage: get_tf_yhat(model) and returns: None, yhat_tf_regression,  str, yhat_tf_classification,  str, yhat_tf_classification,  str