plot_RMAEVI.py

import matplotlib.pyplot as plt

def plot_compare(exp_name, y1_RMAE, y2_RMAE, exp1_id, epx2_id):
  x=["$[0,0.1)$","$[0.1, 0.3)$","$[0.3, 0.5)$","$[0.5, 0.7)$","$[0.7, 0.9)$", r"$[0.9,+\infty)$"]
  noisy_RMAE = [0.293, 0.236, 0.228, 0.226, 0.224, 0.222]
  fig = plt.figure(figsize=(2.7,2.6))
  plt.plot(x, noisy_RMAE, label='Noisy', linewidth=1, color='black', marker='s',
           markerfacecolor='black', markersize=6)
  plt.plot(x, y1_RMAE, label=exp1_id, linewidth=1, color='blue', marker='o',
           markerfacecolor='blue', markersize=6)
  plt.plot(x, y2_RMAE, label=epx2_id, linewidth=1, color='red', marker='>',
           markerfacecolor='red', markersize=6)
  fig.autofmt_xdate()
  plt.xlabel('Value Interval')
  plt.ylabel('RMAE')
  plt.legend()
  plt.subplots_adjust(top=0.98, right=0.96, left=0.23, bottom=0.31)
  plt.savefig("exp/paper_sample/RMAEVI_%s.jpg" % exp_name)
  # plt.show()
  plt.close()

if __name__ == "__main__":

  exp_name = "IRM_ReLU_A10"
  y1_RMAE = [0.105, 0.141, 0.088, 0.068, 0.055, 0.043]
  y2_RMAE = [0.089, 0.131, 0.084, 0.067, 0.054, 0.039]
  exp1_id = "Exp. 1"
  exp2_id = "Exp. 2"
  plot_compare(exp_name, y1_RMAE, y2_RMAE, exp1_id, exp2_id)

  exp_name = "PSM_ReLU_A10"
  y1_RMAE = [0.106, 0.156, 0.09, 0.068, 0.053, 0.04]
  y2_RMAE = [0.094, 0.149, 0.092, 0.071, 0.055, 0.038]
  exp1_id = "Exp. 3"
  exp2_id = "Exp. 4"
  plot_compare(exp_name, y1_RMAE, y2_RMAE, exp1_id, exp2_id)

  exp_name = "IRM_Real_A10"
  y1_RMAE = [0.112, 0.132, 0.081, 0.063, 0.051, 0.039]
  y2_RMAE = [0.091, 0.132, 0.085, 0.069, 0.056, 0.042]
  exp1_id = "Exp. 5"
  exp2_id = "Exp. 6"
  plot_compare(exp_name, y1_RMAE, y2_RMAE, exp1_id, exp2_id)

  exp_name = "PSM_Real_A10"
  y1_RMAE = [0.112, 0.164, 0.103, 0.08, 0.064, 0.049]
  y2_RMAE = [0.091, 0.161, 0.101, 0.079, 0.063, 0.047]
  exp1_id = "Exp. 7"
  exp2_id = "Exp. 8"
  plot_compare(exp_name, y1_RMAE, y2_RMAE, exp1_id, exp2_id)