code: def compare_dist(
      y_pred_mu_1, y_pred_std_1, y_pred_mu_2, y_pred_std_2
  ):
      """return metrics of comparing two gaussians"""
      # https://stats.stackexchange.com/questions/7440/kl-divergence-between-two-univariate-gaussians
      # http://www.allisons.org/ll/MML/KL/Normal/
      # https://stats.stackexchange.com/questions/323069/can-kl-divergence-ever-be-greater-than-1
      kl = np.log(y_pred_std_2 / y_pred_std_1) + (
          np.square(y_pred_std_1)
          - np.square(y_pred_std_2)
          + np.square(y_pred_mu_1 - y_pred_mu_2)
      ) / (2 * np.square(y_pred_std_2))
      return np.mean(kl)
  
file_path: elu-gelu-kernels/code/experiments/02_shallow_experiments/utils.py
imports: import numpy
  import scipy.stats
  import matplotlib.pyplot
  import datetime
label_desc: Use image node_id 5 for calling a global function with example usage: compare_dist(y_pred_mu_1, y_pred_std_1, y_pred_mu_2, y_pred_std_2) and returns: np