code: def mse_loss(y_pred, y):
      """Calculating Mean Square Error Loss function"""
      return tf.reduce_mean(tf.square(y_pred - y))
  
file_path: mlflow/examples/tensorflow/train.py
imports: import tensorflow
  from sklearn.datasets import fetch_california_housing
  import mlflow
  from mlflow.models import infer_signature
label_desc: Use image node_id 1 for calling a global function with example usage: mse_loss(y_pred, y) and returns: tf