env2pfam/algagpt_full/final_metrics.json

{
  "train": {
    "loss": 12.063952380960638,
    "mse": 12.063949584960938,
    "mae": 2.153740167617798,
    "r2": 0.5561336278915405,
    "r2_mean": 0.26438313722610474,
    "r2_median": 0.24722564220428467,
    "r2_min": -0.034403204917907715,
    "r2_max": 0.6671420335769653,
    "cosine_similarity": 0.7164382338523865
  },
  "val": {
    "loss": 14.190956592559814,
    "mse": 14.092447280883789,
    "mae": 2.3967013359069824,
    "r2": 0.46001797914505005,
    "r2_mean": 0.1044716015458107,
    "r2_median": 0.09539571404457092,
    "r2_min": -0.23320329189300537,
    "r2_max": 0.49621617794036865,
    "cosine_similarity": 0.6523082852363586
  },
  "test": {
    "loss": 13.96311264038086,
    "mse": 14.00570297241211,
    "mae": 2.380729913711548,
    "r2": 0.4745151400566101,
    "r2_mean": 0.11885368078947067,
    "r2_median": 0.10473570227622986,
    "r2_min": -0.3345597982406616,
    "r2_max": 0.6163809299468994,
    "cosine_similarity": 0.6655058264732361
  }
}