{
  "train": {
    "loss": 0.009701163846660744,
    "mse": 0.009701163846660744,
    "cosine_loss": 0.38389346545392816,
    "r2": -0.00014269351959228516,
    "mae": 0.07398466765880585,
    "cosine_similarity": 0.6159994006156921,
    "dim_r2_mean": -0.0001436471939086914,
    "dim_r2_min": -0.0009844303131103516,
    "dim_r2_max": -1.1920928955078125e-07
  },
  "val": {
    "loss": 0.009356671385467053,
    "mse": 0.009356671385467053,
    "cosine_loss": 0.3662286400794983,
    "r2": -0.0056684017181396484,
    "mae": 0.07284688949584961,
    "cosine_similarity": 0.6295995712280273,
    "dim_r2_mean": -0.005823422223329544,
    "dim_r2_min": -0.044977784156799316,
    "dim_r2_max": -1.0728836059570312e-06
  },
  "test": {
    "loss": 0.009309758059680462,
    "mse": 0.009309758059680462,
    "cosine_loss": 0.36371852159500123,
    "r2": -0.005517482757568359,
    "mae": 0.07240588963031769,
    "cosine_similarity": 0.6391298174858093,
    "dim_r2_mean": -0.00648270919919014,
    "dim_r2_min": -0.03397870063781738,
    "dim_r2_max": -9.5367431640625e-07
  }
}