env2pfam/algagpt_light/final_metrics.json

{
  "train": {
    "loss": 12.809910557486795,
    "mse": 12.809890747070312,
    "mae": 2.223891258239746,
    "r2": 0.527130126953125,
    "r2_mean": 0.22765445709228516,
    "r2_median": 0.2012607455253601,
    "r2_min": -0.25727713108062744,
    "r2_max": 0.5581326484680176,
    "cosine_similarity": 0.6956151127815247
  },
  "val": {
    "loss": 14.251752376556396,
    "mse": 14.14225959777832,
    "mae": 2.4045510292053223,
    "r2": 0.4581093192100525,
    "r2_mean": 0.08806457370519638,
    "r2_median": 0.08397451043128967,
    "r2_min": -0.20091509819030762,
    "r2_max": 0.4599531292915344,
    "cosine_similarity": 0.6521943211555481
  },
  "test": {
    "loss": 14.090509510040283,
    "mse": 14.135997772216797,
    "mae": 2.4154062271118164,
    "r2": 0.46962660551071167,
    "r2_mean": 0.10699501633644104,
    "r2_median": 0.09248968958854675,
    "r2_min": -0.603492021560669,
    "r2_max": 0.5688364505767822,
    "cosine_similarity": 0.6636574268341064
  }
}