{
  "train": {
    "loss": 0.009715628852559761,
    "mse": 0.009715628852559761,
    "cosine_loss": 0.3846368572928689,
    "r2": -8.52346420288086e-05,
    "mae": 0.07396243512630463,
    "cosine_similarity": 0.6160224080085754,
    "dim_r2_mean": -8.874572813510895e-05,
    "dim_r2_min": -0.0004864931106567383,
    "dim_r2_max": -1.1920928955078125e-07
  },
  "val": {
    "loss": 0.00936005897819996,
    "mse": 0.00936005897819996,
    "cosine_loss": 0.3663846731185913,
    "r2": -0.006044507026672363,
    "mae": 0.07282450050115585,
    "cosine_similarity": 0.6294302940368652,
    "dim_r2_mean": -0.006076883524656296,
    "dim_r2_min": -0.048920631408691406,
    "dim_r2_max": -2.5033950805664062e-06
  },
  "test": {
    "loss": 0.009311655163764953,
    "mse": 0.009311655163764953,
    "cosine_loss": 0.36379417181015017,
    "r2": -0.0057315826416015625,
    "mae": 0.07238934934139252,
    "cosine_similarity": 0.6390543580055237,
    "dim_r2_mean": -0.00670330785214901,
    "dim_r2_min": -0.032412052154541016,
    "dim_r2_max": -2.9802322387695312e-06
  }
}