env2pfam/pythia_light/final_metrics.json

{
  "train": {
    "loss": 14.185515382073142,
    "mse": 14.185503005981445,
    "mae": 2.41489839553833,
    "r2": 0.41983503103256226,
    "r2_mean": 0.19313760101795197,
    "r2_median": 0.14171487092971802,
    "r2_min": -0.18446409702301025,
    "r2_max": 0.46663594245910645,
    "cosine_similarity": 0.6183400750160217
  },
  "val": {
    "loss": 14.829891395568847,
    "mse": 14.853211402893066,
    "mae": 2.4977495670318604,
    "r2": 0.41088223457336426,
    "r2_mean": 0.16360466182231903,
    "r2_median": 0.11254703998565674,
    "r2_min": -0.6352555751800537,
    "r2_max": 0.4856938123703003,
    "cosine_similarity": 0.6161171197891235
  },
  "test": {
    "loss": 14.504690551757813,
    "mse": 14.561346054077148,
    "mae": 2.4540226459503174,
    "r2": 0.40021586418151855,
    "r2_mean": 0.143249973654747,
    "r2_median": 0.10639423131942749,
    "r2_min": -0.14785397052764893,
    "r2_max": 0.4375695586204529,
    "cosine_similarity": 0.6081491708755493
  }
}