env2pfam/pythia_full/final_metrics.json

{
  "train": {
    "loss": 13.159953225742687,
    "mse": 13.159968376159668,
    "mae": 2.26400089263916,
    "r2": 0.46306395530700684,
    "r2_mean": 0.2584795653820038,
    "r2_median": 0.20972812175750732,
    "r2_min": 0.03256005048751831,
    "r2_max": 0.5723539590835571,
    "cosine_similarity": 0.6514576077461243
  },
  "val": {
    "loss": 14.689822673797607,
    "mse": 14.72309398651123,
    "mae": 2.451831579208374,
    "r2": 0.4160430431365967,
    "r2_mean": 0.17224988341331482,
    "r2_median": 0.12200796604156494,
    "r2_min": -0.39256393909454346,
    "r2_max": 0.5682699680328369,
    "cosine_similarity": 0.6213155388832092
  },
  "test": {
    "loss": 14.533664798736572,
    "mse": 14.596628189086914,
    "mae": 2.411353588104248,
    "r2": 0.398762583732605,
    "r2_mean": 0.14865508675575256,
    "r2_median": 0.10728722810745239,
    "r2_min": -0.18526697158813477,
    "r2_max": 0.566988468170166,
    "cosine_similarity": 0.6101461052894592
  }
}